如何在列表，字典，集合中根据条件筛选数据
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()根据判断结果自动过滤掉不符合条件的元素，返回由符合条件元素组成的新listfrom random import randintdata = [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 randintdata = [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])