Python学习29:filter
Python内建的filter()
函数用于过滤序列。
和map()
类似,filter()
也接收一个函数和一个序列。和map()
不同的是,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。
例如,在一个list中,删掉偶数,只保留奇数,可以这么写:
def is_odd(n):
return n % 2 == 1
list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
# 结果: [1, 5, 9, 15]
把一个序列中的空字符串删掉,可以这么写:
def not_empty(s):
return s and s.strip()
list(filter(not_empty, ['A', '', 'B', None, 'C', ' ']))
# 结果: ['A', 'B', 'C']
可见用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
开始的奇数序列:
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) # 构造新序列
这个生成器先返回第一个素数2
,然后,利用filter()
不断产生筛选后的新的序列。
由于primes()
也是一个无限序列,所以调用时需要设置一个退出循环的条件:
# 打印1000以内的素数:
for n in primes():
if n < 1000:
print(n)
else:
break
注意到Iterator
是惰性计算的序列,所以我们可以用Python表示“全体自然数”,“全体素数”这样的序列,而代码非常简洁。
小结
filter()
的作用是从一个序列中筛出符合条件的元素。由于filter()
使用了惰性计算,所以只有在取filter()
结果的时候,才会真正筛选并每次返回下一个筛出的元素。
更多内容,可关注作者的微信公众号:胖哥真不错。
相关文章
- Python快速学习07:文本文件的操作
- 关于Python正则表达式findall函数问题详解
- 机器学习之决策树(ID3)算法与Python实现
- 机器学习笔记 - 使用plotnine在python中实现ggplot
- NLP之ASR:基于python和机器学习算法带你玩转的语音实时识别技术
- Python:利用python语言实现18位身份证号码和15位身份证号码相互转换
- 〖Python零基础入门篇㉗〗- Python中集合的常用方法
- 最棒的机器学习(Python代码实现)
- python输入字符串,统计字母和数字有多少
- 深度学习“四大名著”发布!Python、TensorFlow、机器学习、深度学习四件套(附免费下载)
- Python学习103:使用模板
- Python学习75:itertools
- Python项目学习二:项目2:绘制图表--再次实现
- 【华为OD机试 2023】租车骑绿岛(C++ Java JavaScript Python)
- Python编程:使用decimal定点数计算0.1+0.2
- [Python 学习] 两、在Linux使用平台Python
- python 反扒技巧 爬虫失败的原因分析 html的学习 css java html dom bs4 xpath selenium js混淆加密 反混淆 ajaxs
- Python正则表达式
- python 3.6 + robotFramework自动化框架 环境搭建、学习笔记
- 调用另一个文件的python代码【转载】
- 爬虫学习之-Python list 和 str 互转
- python学习之基本语法---语法规则---注释,标识符,关键字,命名规则(二)day8
- 机器学习:Python实现聚类算法(三)之总结
- 画【Python折线图】的一百个学习报告(三、设置全局 Label 颜色)