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

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

1. 过滤类别中的负数
# eg_v1 使用迭代判断

data = [1,4,5,-1,-2,-5,9,10,11]
res1 = []
for x in data:
if x >= 0:
res1.append(x)
print (res1)
# [1, 4, 5, 9, 10, 11]
res2 = []
for y in data:
if y < 0:
res2.append(y)
print (res2)
# [-1, -2, -5]

　　

# eg_v2 使用filter函数
filter()函数接收一个函数 f 和一个list，这个函数 f 的作用是对每个元素进行判断，返回 True或 False，filter()根据判断结果自动过滤掉不符合条件的元素，返回由符合条件元素组成的新list

from random import randint
data = [randint(-10,10) for _ in xrange(10)]
print (data)
# [-8, 8, 0, -9, -9, 8, -10, 9, -4, 5]
F1 = filter(lambda x : x >= 0,data)
print (F1)
# [8, 0, 8, 9, 5]

　　

# eg_v3 使用列表解析

from random import randint
data = [randint(-10,10) for _ in xrange(10)]
print (data)
# [-3, -4, -9, 1, -5, 6, -7, -6, 10, 7]
F2 = [x for x in data if data >= 0]
print (F2)
# [1, 6, 10, 7]

　　

可以使用timeit 函数测试小段代码片段的执行时间

T1 = timeit filter(lambda x : x >= 0,data)
T2 = timeit [x for x in data if data >= 0]

　　

2. 筛选字典中的值
# eg_v4 创建一个字典，并筛选值大于80的数

d = {x: randint(60,100) for x in range(1,20)} 
print (d)
# {1: 93, 2: 98, 3: 70, 4: 80, 5: 60, 6: 73, 7: 100, 8: 71, 9: 86, 10: 87, 11: 62, 12: 97, 13: 87, 14: 96, 15: 77, 16: 70, 17: 79, 18: 87, 19: 99}
dict1 = {k: v for k, v in d.iteritems() if v > 80} # k 为字典中的键，v 为字典中的值
print (dict1)
# {1: 93, 2: 98, 7: 100, 9: 86, 10: 87, 12: 97, 13: 87, 14: 96, 18: 87, 19: 99}

　　

3. 筛选集合中的元素
# eg_v5 筛选出集合能被3整除的元素

set1 = {12,-1,3,4,7,8,23,18,23,11}
s1 = {x for x in set1 if x % 3 == 0}
print s1
# set([18, 3, 12])