本文使用python 3.5.2环境
经常会遇到如下需求:
- 过滤列表
[5, 18, -9, 35, 20, -12 ...]
中的负数 - 筛出字典
{'Andrew':64, 'Jack':88, 'Leo':96 ...}
中值高于80的项 - 筛出集合
{54, 65, 37, 43 ...}
中能被3整除的元素
通常的做法是迭代,例如:
data = [3, 6, -2, 9, 4, 0, 7]
data_new = []
for i in data:
if i >= 3:
data_new.append(i)
print('data_new')
这样使用显得比较繁琐,我们可以使用一些高级函数来简化一下。
列表
列表筛选数据有filter函数和列表生成式两种。
这里生成一个范围在-10到10之间的10个元素的列表,然后筛选出其中大于等于0的元素:
from random import randint
data = [randint(-10, 10) for i in range(1, 11)]
print(data) #[5, 7, 9, 3, -6, -7, -8, -9, 3, -8]
new1 = list(filter(lambda x: x >= 0, data))
print(new1) #用filter函数的方法筛选,结果为[5, 7, 9, 3, 3]
new2 = [x for x in data if x >= 0]
print(new2) #用列表生成式的方法筛选,结果和filter筛选结果一致
注:用列表生成式筛选速度比filter函数要快,可以用timeit来验证。
字典
这里我们生成一个字典,代表着20个人编号为1~20,每个人成绩在60~100分随机生成,筛选出成绩大于90分的学生编号。
from random import randint
d = {x: randint(60, 100) for x in range(1, 21)}
d1 = {k: v for k, v in d.items() if v > 90}
集合
我们把前面的data转为集合,筛选出能被3整除的元素
s = set(data)
{x for x in s if x % 3 == 0}