python高阶函数—filter

时间:2023-01-30 13:31:03

python内置了一个filter函数,用于过滤序列。和map函数类似,filter()函数也接受一个函数和一个序列。只不过filter函数中是把函数依次作用于序列中的每一个元素,如果是True则保留这个元素,如果是False,则舍弃这个元素。例如,给定一个list,删除偶数,保留奇数:

>>> def is_odd(n):
...     return n % 2 ==1
...
>>> list(filter(is_odd,[1,2,3,4,5,6]))
[1, 3, 5]

注意,filter返回的是一个Iterator,俗称惰性序列,所以要使用list()函数获得所有元素返回一个list。

用filter求素数:

素数的定义:又称质数,为大于1的自然数中,除了1和它本身以外不再有其他因数。

计算素数的一个方法是埃氏筛法:

首先从2开始的所有自然数中:

2,3,4,5,6,7,8,9……

取序列的第一个数2,他一定是素数,然后用2把序列的倍数去掉:

3,5,7,9……

去新序列的第一个数3,他一定是素数,然后用3 把序列的3的倍数去掉:

5,7……:

5一定是素数,然后用5把序列的5的倍数去掉:

7,11……

程序如下:

>>> def odd_iter():#构造一个从3开始的奇数序列
...     n=1
...     while True:
...             n = n+2
...             yield n
...
>>> def not_divisible(n):#筛选函数
...     return lambda x:x%n > 0
...
>>> def primes(): #生成器,不断返回下一个素数
...     yield 2
...     it = odd_iter()#初始化序列
...     while True:
...             n = next(it)#取序列的第一个数
...             yield n
...             it = filter(not_divisible(n),it)#构造新序
...
>>> for n in primes():
...     if n < 1000:
...             print(n)
...     else:
...             break
...

结果:

2
3
5
7
11
13
17
19
23
……
953
967
971
977
983
991
997

练习:

利用filter函数筛选出1~1000中的回数(回数是指从左往右和从右到左读都一样的数,例如1221,676):

>>> def is_palindrome(n):
...     return n == int(str(n)[::-1])
...
>>> list(filter(is_palindrome,range(1,1000)))
[1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 22, 33, 44, 55, 66, 77, 88,
……
39, 949, 959, 969, 979, 989, 999]