python的filter函数

柔光的暖阳◎ 2021-11-27 01:28 266阅读 0赞

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

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

例如，在一个list中，删掉偶数，只保留奇数，可以这么写：  
![在这里插入图片描述][20190723143132160.png]  
把一个序列中的空字符串删掉，可以这么写：  
![在这里插入图片描述][2019072314360173.png]  
可见用`filter()`这个高阶函数，关键在于正确实现一个“筛选”函数。

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

**用filter求素数**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjQxOTU3_size_16_color_FFFFFF_t_70]

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

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

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

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

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

### 小结 ###

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

原文地址：[https://www.liaoxuefeng.com/wiki/1016959663602400/1017404530360000][https_www.liaoxuefeng.com_wiki_1016959663602400_1017404530360000]

[20190723143132160.png]: /images/20211126/e0c237ca60e3434cacee459c35d743e1.png
[2019072314360173.png]: /images/20211126/f5ff99c80265426aacdd7bbd2a7f7c1c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwMjQxOTU3_size_16_color_FFFFFF_t_70]: /images/20211126/f67be959be5b429c8f68e9d6c787e219.png
[201907231458273.png]: /images/20211126/f1c48562120b445bbb0334e4b53ca0a7.png
[https_www.liaoxuefeng.com_wiki_1016959663602400_1017404530360000]: https://www.liaoxuefeng.com/wiki/1016959663602400/1017404530360000