Python内建的filter()函数用于过滤序列。filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素。返回由符合条件元素组成的新列表。
例子:
def is_odd(n):
return n % 2 == 1
filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15])
结果: [1, 5, 9, 15]
import math
def is_sqr(x):
return math.sqrt(x) % 1 == 0
newlist = list(filter(is_sqrt, range(1, 101))) #注意用list封装了下
print(newlist)
输出结果:[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
当出现这种错误时,是因为没将filter函数转换成list。
import math
def is_sqr(x):
return math.sqrt(x) % 1 == 0
newlist = filter(is_sqrt, range(1, 101))
print(newlist)
结果:
注意:关于filter()方法, python3和python2有一点不同
python2中返回的是过滤后的列表, 而python3中返回到是一个filter类
filter类实现了iter和next方法, 可以看成是一个迭代器, 有惰性运算的特性, 相对python2提升了性能, 可以节约内存.