内置函数filter
filter() 函数用于过滤序列,过滤掉不符合条件的元素,返回由符合条件元素组成的新列表。
该函数接收两个参数,第一个为函数,第二个为序列,序列的每个元素作为参数传递给函数进行判,然后返回 True 或 False,最后将返回 True 的元素放到新列表中。
例:
list0 = [1,2,3,4,5,6,7,8,9] def remove_odd(x): return x%2 != 1 #去除奇数 print(list(filter(remove_odd,list0))) import math #调用模块 def square(x): #留下平方根是整数的数字 return math.sqrt(x) % 1 == 0 print(list(filter(square,list0)))
[2, 4, 6, 8]
[1, 4, 9]