Python内建的filter()函数用于过滤序列。
filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素
1.求素数的例子:
①先构造一个从3开始的奇数序列:
def _odd_iter():
n = 1
while True:
n = n + 2
yield n
②定义一个筛选函数
a = lambda x: x % 10 > 0
现在a就是一个函数
a(1) 就等价于 1 % 10 > 0
最后返回结果
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) # 构造新序列
这个生成器先返回第一个素数2,然后,利用filter()不断产生筛选后的新的序列。
由于primes()也是一个无限序列,所以调用时需要设置一个退出循环的条件:
打印1000以内的素数:
for n in primes():
if n < 1000:
print(n)
else:
break
注意到Iterator是惰性计算的序列,所以我们可以用Python表示“全体自然数”,“全体素数”这样的序列,而代码非常简洁。
filter()的作用是从一个序列中筛出符合条件的元素。由于filter()使用了惰性计算,所以只有在取filter()结果的时候,才会真正筛选并每次返回下一个筛出的元素。
例子:
#!/usr/bin/python
# -*- coding: UTF-8 -*-
'''
回数是指从左向右读和从右向左读都是一样的数,
例如12321,909。请利用filter()滤掉非回数:
'''
def is_palindrome(n):
#把整形转变成为字符串
n=str(n)
a=len(n)/2
if n.rfind(n[0:int(a)],len(n)-int(a),len(n))>-1:
return n
else:
pass
# 测试:
output = filter(is_palindrome, range(1, 1000))
print(list(output))
更简单:
def is_palidrome(n):
return str(n)==str(n)[::-1]