python基础——filter函数
python基础——filter函数
Python内建的filter()
函数用于过滤序列。
和map()
类似,filter()
也接收一个函数和一个序列。和map()
不同的是,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。
例如,在一个list中,删掉偶数,只保留奇数,可以这么写:
#在一个list中,删掉偶数,只保留奇数 def is_odd(n): return n%2==1 L1=filter(is_odd,[1,2,3,4,5,6,7,8,9,10,15]) print(list(L1)) #输出:[1,3,5,7,9,15]
把一个序列中的空字符串删掉,可以这么写:
#把一个序列的空字符串删掉 def not_empty(s): return s.strip() L2=filter(not_empty, ['A', '', 'B', 'C', ' ']) print(list(L2)) #输出结果:['A','B','C'] ''' strip()函数说明: ---------------------------------------------------------------------- s.strip(rm) 删除s字符串中开头、结尾处,位于 rm删除序列的字符 s.lstrip(rm) 删除s字符串中开头处,位于 rm删除序列的字符 s.rstrip(rm) 删除s字符串中结尾处,位于 rm删除序列的字符 注意: 1. 当rm为空时,默认删除空白符(包括'\n', '\r', '\t', ' ') ----------------------------------------------------------------------- '''
可见用filter()
这个高阶函数,关键在于正确实现一个“筛选”函数。
注意到filter()
函数返回的是一个Iterator
,也就是一个惰性序列,所以要强迫filter()
完成计算结果,需要用list()
函数获得所有结果并返回list。
用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, ...
不断筛下去,就可以得到所有的素数。
用Python来实现这个算法,可以先构造一个从3
开始的奇数序列:
#用filter求素数 #可以先构造一个从3开始的奇数序列:(从3开始的偶数序列都不是素数) def _odd_iter(): n=1 while True: n+=2 yield n #构造一个生成器,并且是一个无限序列
注意这是一个生成器,并且是一个无限序列。
然后定义一个筛选函数:
#定义一个筛选函数 def _not_divisible(n): return lambda x:x%n >0 #Lambda表示生成一个匿名函数 #带有一个参数的Lambda表达式和对应的非匿名函数: #举例: # def f(n): # return lambda x: x / n #print ( f(1)(2) ) # n=1; x=2
最后,定义一个生成器,不断返回下一个素数:
#最后,定义一个生成器,不断返回下一个素数 def primes(): yield 2 #先返回第一个素数2,然后利用filter()函数不断产生筛选后的新序列 it=_odd_iter() while True: n=next(it) #取出序列的第一个数 yield n #返回序列的第一个数 it=filter(_not_divisible,it) #此处用_not_divisible()函数去作用it列表中的每一个数(设为x), #则函数实际的模样为_not_divisible(n)(x),函数所执行的操作为:x%n >0,当可以整除时,丢弃,当不能整除时,筛选出来
这个生成器先返回第一个素数2
,然后,利用filter()
不断产生筛选后的新的序列。
由于primes()
也是一个无限序列,所以调用时需要设置一个退出循环的条件:
#打印1000以内的素数: def main(): for n in primes(): if n<1000: print(n) else: break if __name__=='__main__': main() ''' 如果我们是直接执行某个.py文件的时候,该文件中那么”__name__ == '__main__'“是True, 但是我们如果从另外一个.py文件通过import导入该文件的时候, 这时__name__的值就是我们这个py文件的名字而不是__main__。 这个功能还有一个用处:调试代码的时候,在”if __name__ == '__main__'“中加入一些我们的调试代码, 我们可以让外部模块调用的时候不执行我们的调试代码,但是如果我们想排查问题的时候, 直接执行该模块文件,调试代码能够正常运行 '''
注意到Iterator
是惰性计算的序列,所以我们可以用Python表示“全体自然数”,“全体素数”这样的序列,而代码非常简洁。
练习
回数是指从左向右读和从右向左读都是一样的数,例如12321
,909
。请利用filter()
滤掉非回数:
#练习 ''' 回数是指从左向右读和从右向左读都是一样的数,例如12321,909。 请利用filter()滤掉非回数: ''' def is_palindrome(n): return str(n)==str(n)[::-1] #将输入的数字转为字符串,将翻转前和翻转后的进行比较,如果一致,表示满足回数 print('1-15000的回数有:') print(list(filter(is_palindrome,range(1,15000))))
运行结果:
小结
filter()
的作用是从一个序列中筛出符合条件的元素。由于filter()
使用了惰性计算,所以只有在取filter()
结果的时候,才会真正筛选并每次返回下一个筛出的元素。