filter
Python内建的filter()
函数用于过滤序列。
filter()
也接收一个函数和一个序列。filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。
例1、在一个list中,删掉奇数,只保留偶数
def is_odd(n):
if n%2 ==0:
return n
L = list(range(1,11))
print('L:',L)
print(list(filter(is_odd,L)))
输出:
L: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
[2, 4, 6, 8, 10]
例2、用filter求素数
思路:
首先,列出从2
开始的所有自然数,构造一个序列:
2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取序列的第一个数2
,它一定是素数,然后用2
把序列的2
的倍数筛掉:
3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取新序列的第一个数3
,它一定是素数,然后用3
把序列的3
的倍数筛掉:
5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
取新序列的第一个数5
,然后用5
把序列的5
的倍数筛掉:
7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, …
不断筛下去,就可以得到所有的素数。
#先构造一个从3开始的奇数序列
#注意这是一个生成器,并且是一个无限序列。
def _odd_iter():
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) # 构造新序列
# 打印20以内的素数:
for n in primes():
if n < 20:
print(n)
else:
break
输出:
2
3
5
7
11
13
17
19