如何在列表、字典、集合中根据条件筛选数据
列表过滤负数
data = [-1,2,3,-4,5]
普通方式 使用for循环if判断进行过滤
res = []
for i in data:
if i >= 0:
res.append(i)
print(res)
高级
1. 列表解析
2. filter函数
## 列表解析
data = [-1, 2, 3, -4, 5]
data = [x for x in data if x >= 0]
data
## filter函数【过滤函数filter(函数,可迭代对象)】
```python
data = [-1, 2, 3, -4, 5]
data = filter(lambda x: x >= 0, data)
list(data)
筛选字典中满足条件的数据
需求:字典解析过滤出成绩大于90分的同学
# 字典过滤器
from random import *
stu = {'student%d' %i:randint(50,100) for i in range(1,21)}
stu1 = {k:v for k,v in stu.items() if v>=90}
stu1
# filter函数过滤
stu2 = filter(lambda item: item[1] >= 90, stu.items())
dict(stu2)
集合:被3整除的数
s = {randint(0, 20) for _ in range(20)}
c = {x for x in s if x % 3 == 0}
c