Python如何在列表、字典、集合中根据条件筛选数据

列表

>>> data = [x for x in range(10)]
>>> data
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

筛选出能被二整除的数

• 最通用的做法当然是循环+判断

>>> l = []
>>> for i in range(10):
...     if i % 2 == 0:
...         l.append(i)
...
>>> l
[0, 2, 4, 6, 8]
>>>

但是这种方法比较繁琐而且效率低下，可以使用一些其他的高级的方法

• filter

>>> f = filter(lambda x: x % 2 == 0, range(10))
>>> f

>>> for i in f:
...     print(i)
...
0
2
4
6
8

filter函数接受两个参数，第一个是定义的过滤函数，第二个是一个序列，返回一个惰性序列，通过迭代可以得到值

• 列表解析

>>> [i for i in range(10) if i % 2 == 0]
[0, 2, 4, 6, 8]

这种方法是最常用的方法也是效率最高的方法，在过滤规则简单的时候推荐使用

字典

>>> from random import randint
>>> data = {no: randint(40, 100) for no in range(1, 21)}
>>> data
{1: 54, 2: 65, 3: 69, 4: 44, 5: 60, 6: 54, 7: 52, 8: 84, 9: 95, 10: 77, 11: 51, 12: 61, 13: 69, 14: 44, 15: 63, 16: 67, 17: 86, 18: 59, 19: 83, 20: 59}
>>>

下面要筛选出出成绩大于80分的学生

• 字典解析

>>> {k: v for k, v in data.items() if v >= 80}
{8: 84, 9: 95, 19: 83, 17: 86}

集合

>>> data = set([x for x in range(10)])
>>> data
{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}

筛选出能被三整除的数

• 集合解析

>>> {x for x in data if x % 3 == 0}
{0, 9, 3, 6}