文章最后更新时间为:2018年08月14日 11:15:45
这篇就录怎么在列表集合字典中根据需要筛选数据。
1. 筛选出数字列表中的正数
假如我们有一个数字列表,其中的数字有正有负,这时候如果我们只想要其中的正数,那么我们怎么样快速得到去除负数后的列表呢?
一般我们都会想要最简单但是也是最耗时的方法--迭代.
首先我们创造一个随机列表:>>> from random import randint
>>> data=[randint(-10,10) for _ in range(10)]
>>> data
[2, -6, 6, 4, 10, -5, 6, 9, -8, -6]
然后迭代去除其中的负数:>>> data1=[]
>>> for i in data:
... if(i>=0):
... data1.append(i)
...
>>> data1
[2, 6, 4, 10, 6, 9]
可以看出这时已经成功了,但是这种方式比较耗时耗力。
我们还可以有种更简单的方法也就是使用过滤函数filter
看下列代码:>>> data3=list(filter(lambda x:x>=0,data))
>>> data3
[2, 6, 4, 10, 6, 9]
使用filter()函数可以过滤掉不符合x>=0的数,在python3中filter()返回的是一个filter类,所以我们需要使用list函数将其转化为一个列表。
第三种方法也是最快的方法就是直接在列表中过滤,看下面代码:>>> data3=[x for x in data if x>=0 ]
>>> data3
[2, 6, 4, 10, 6, 9]
上述三种方法一般是第三种方法最快速,第一种方法最慢。
代码块:from random import randint
data=[randint(-10,10) for _ in range(10)]
data1=[]
for i in data:
if(i>0):
data1.append(i)
data2=list(filter(lambda x: x>0,data))
data3=[x for x in data if x>0 ]
print(data)
print("data1:")
print(data1)
print("data2:")
print(data2)
print("data3:")
print(data3)
2. 筛选出字典中的特定键值对
假设我们有了这样一个字典,键是学生的学号,值是学生的考试成绩。这时我们想要查看90分以上的情况,所以需要将值>=90的筛选出来。
首先创造这个成绩字典:>>> data={x:randint(50,100) for x in range(10)}
>>> data
{0: 95, 1: 51, 2: 100, 3: 89, 4: 95, 5: 88, 6: 80, 7: 69, 8: 50, 9: 99}
然后我们依然可以利用列表的方法将其中值大于90的键值对找出来:>>> data1={k:v for k,v in data.items() if v>90}
>>> data1
{0: 95, 2: 100, 4: 95, 9: 99}
3. 筛选出集合中的3的倍数
方法和列表类似,直接看代码:>>> from random import randint
>>> data={randint(-10,10) for _ in range(25)}
>>> data
{2, 3, 6, 7, 8, 9, -10, -9, -8, -7, -6, -5, -4, -2}
>>> data1={x for x in data if x%3==0}
>>> data1
{3, 6, 9, -9, -6}