python列表，字典和集合筛选数据方法

本文使用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}