如何在列表、字典、集合中根据条件筛选数据
2017-11-13 本文已影响0人
万越天
迭代方式(Low)
data = [1, 5, -3, -2, 6, 0, 9]
res = []
for x in data:
if x >= 0:
res.append(x)
print(res)
列表:过滤掉列表中的负数
from random import randint
data = [randint(-10, 10) for _ in range(10)]
# 方式一 filter + lambda
res = filter(lambda x: x >= 0, data) # python3 中为迭代器,需要for循环进行遍历
for item in res:
print(item)
# 方式二 列表生成式(推荐)
res = [x for x in data if x >= 0]
print(res)
字典:过滤字典中值大于90的值
from random import randint
data = {x: randint(0, 100) for x in range(1, 21)}
res = {k: v for k, v in data.items() if v > 90}
集合:过滤集合中能被3整除的元素
from random import randint
data = [randint(-10, 10) for _ in range(10)]
s = set(data)
res = {x for x in s if x % 3 == 0}
print(res)