filter
函数可以用来过滤序列中不符合条件的元素,只保留符合条件的元素,第一个参数为过滤条件(函数)返回bool
型变量,True或者False,第二个参数为序列,例如过滤序列中小于0.8的元素。
import numpy as np
filter_list = filter(lambda x: x > .8, np.random.rand(10))
print(list(filter_list))
>>> [0.8199839816895357, 0.8437691488645747, 0.8588727113271978, 0.8132885456164843]