Python的filter函数

时间:2025-02-16 09:46:26

filter()函数是 Python 内置的另一个有用的高阶函数,filter()函数接收一个函数 f 和一个序列,函数 f 的作用是对每个元素进行判断,返回 True或 False,filter()根据判断结果自动过滤掉不符合条件的元素,返回由符合条件元素组成的新的序列Iterator。

例如,要从一个list [1, 4, 6, 7, 9, 12, 17]中删除偶数,保留奇数,首先,要编写一个判断奇数的函数:

def is_odd(x):
    return x % 2 == 1

然后,利用filter()过滤掉偶数:

r = filter(is_odd, [1, 4, 6, 7, 9, 12, 17])
print(list(r))

 结果:[1, 7, 9, 17]

利用filter(),可以完成很多有用的功能,例如,删除 None 或者空字符串:

def is_not_empty(s):
    return s and len(()) > 0  #或者直接return s and ()
r = filter(is_not_empty, ['test', None, '', 'str', '  ', 'END'] 
print(list(r))

结果:['test', 'str', 'END']

 注意到 filter()函数返回的是一个 Iterator,也就是一个惰性序列,所以要强迫 filter()完成计算结果,需要用 list()函数获得所有结果并返回 list。

注:可见用 filter()这个高阶函数,关键在于正确实现一个“筛选”函数。

实例1. 请利用filter()过滤出1~100中平方根是整数的数,即结果应该是:

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

import math
def is_sqr(x):
    r = int((x))
    return r * r == x
t = filter(is_sqr, range(1, 101))
print(list(t))
#输出结果
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

实例2. 用 filter() 求素数、

待完成。。。。。。