Python高级编程技巧实战（1）如何在列表、字典、集合中根据条件筛选数据

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

实际案例：

案例1：过滤掉列表[3, 9, -1, 10, 20, -2, ...] 中的负数

案例2：筛出字典{'lisi': 79, 'Jin': 88, 'lucy': 93, ... }中值高于90的项

案例3：筛出集合{77, 89, 34, 20, 21...}中能被3整除的元素

这类问题比较简单，通常的做法就是依次迭代列表、字典、集合中的每个项，进行条件判断。

但是在python中，还有更高级的方法来解决这类问题，并且更简单高效。

案例1：过滤掉列表[3, 9, -1, 10, 20, -2, ...] 中的负数

方法一：使用 filter 函数

from random import randint

# 使用列表解析生成 -10~10 之间的10个元素
data = [randint(-10, 10) for _ in range(10)]

print('原始列表为：'data)

# filter(function or None, iterable) --> filter object
data_o = filter(lambda x: x >= 0, data)

for each in data_o:
    
    print(each)

方法二：使用列表解析

from random import randint

# 使用列表解析生成 -10~10 之间的10个元素
data = [randint(-10, 10) for _ in range(10)]

print('原始列表为：'data)

data_o = [x for x in data if x >= 0]

print(data_o)

案例2：筛出字典{'lisi': 79, 'Jin': 88, 'lucy': 93, ... }中值高于90的项

from random import randint

# 使用字典解析生成 一个字典
d ={x: randint(60, 100) for x in range(1, 10)}
print(d)

d_o = {k: v for k, v in d.items() if v >= 90}

print(d_o)

案例3：筛出集合{77, 89, 34, 20, 21...}中能被3整除的元素

from random import randint

# 使用集合解析生成 -10~10 之间的10个元素
data = {randint(-10, 10) for _ in range(10)}

print('原始集合为：'data)

data_o = {x for x in data if x % 3 == 0}

print(data_o)