python的filter函数

最新推荐文章于 2025-06-05 09:45:48 发布

转载最新推荐文章于 2025-06-05 09:45:48 发布 · 8.2k 阅读

文章标签：

44 篇文章

订阅专栏

本文介绍了Python内置函数filter()的工作原理及其应用场景。filter()通过指定的函数从序列中筛选出符合条件的元素，返回一个惰性计算的迭代器。文章详细解释了如何使用filter()结合lambda表达式来删除序列中的特定元素，并提供了一个生成素数的例子。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python内建的filter()函数用于过滤序列。

和map()类似，filter()也接收一个函数和一个序列。和map()不同的是，filter()把传入的函数依次作用于每个元素，然后根据返回值是True还是False决定保留还是丢弃该元素。

例如，在一个list中，删掉偶数，只保留奇数，可以这么写：
在这里插入图片描述
把一个序列中的空字符串删掉，可以这么写：

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

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

用filter求素数
在这里插入图片描述

def _odd_iter():
    n = 1
    while True:
        n = n + 2
        yield n

注意这是一个生成器，并且是一个无限序列。

然后定义一个筛选函数：

def _not_divisible(n):
    return lambda x: x % n > 0      //这句话的意思是if x%n>0 return x

最后，定义一个生成器，不断返回下一个素数：
在这里插入图片描述
这个生成器先返回第一个素数2，然后，利用filter()不断产生筛选后的新的序列。

由于primes()也是一个无限序列，所以调用时需要设置一个退出循环的条件：

# 打印1000以内的素数:
for n in primes():
    if n < 1000:
        print(n)
    else:
        break

注意到Iterator是惰性计算的序列，所以我们可以用Python表示“全体自然数”，“全体素数”这样的序列，而代码非常简洁。

filter()的作用是从一个序列中筛出符合条件的元素。由于filter()使用了惰性计算，所以只有在取filter()结果的时候，才会真正筛选并每次返回下一个筛出的元素。