python从列表字典集合中筛选数据

文章最后更新时间为：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}