python 函数式编程之高阶函数filter

python学习笔记,特做记录,分享给大家,希望对大家有所帮助。

高阶函数filter

Python内建的filter()函数用于过滤序列。

和map()类似,filter()也接收一个函数和一个序列。和map()不同的是,filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素。

例如,在一个list中,删掉偶数,只保留奇数,可以这么写:

def is_odd(n):
    return n % 2 == 1

print list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))

运行结果:

[1, 5, 9, 15]

Process finished with exit code 0

把一个序列中的空字符串删掉,可以这么写:

def not_empty(s):
    return s and s.strip()

print list(filter(not_empty, ['A', '', 'B', None, 'C', '   ']))

运行结果:

['A', 'B', 'C']

Process finished with exit code 0

可见用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 m in primes():
    if m < 1000:
        print m
    else:
        break

注意到Iterator是惰性计算的序列,所以我们可以用Python表示“全体自然数”,“全体素数”这样的序列,而代码非常简洁。

欢迎关注公众号「网罗开发」,可领取python测试demo和学习资源,大家一起学python,网罗天下方法,方便你我开发

希望可以帮助大家,如有问题可加QQ技术交流群: 668562416
如果哪里有什么不对或者不足的地方,还望读者多多提意见或建议
如需转载请联系我,经过授权方可转载,谢谢

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 直接上正文 函数是Python内建支持的一种封装,我们通过把大段代码拆成函数,通过一层一层的函数调用,就可以把复杂...
    OzanShareing阅读 3,325评论 0 0
  • 文章来源 基本是拷贝的内容,目的是为了将知识点整理在一起。除了一个小结(装饰器部分)提供了我自己的解法,其余基本没...
    王诗翔阅读 3,868评论 0 1
  • 函数是Python内建支持的一种封装,我们通过把大段代码拆成函数,通过一层一层的函数调用,就可以把复杂任务分解成简...
    祐吢房_2c9a阅读 3,110评论 1 1
  • 早餐:一小碗燕麦,快走40分钟,计4000步 午餐:大半碗青菜面条,一盘拌豆腐,快走20分钟,计2000步 晚餐:...
    吃窝边草的兔子阅读 1,175评论 0 1
  • 我记得我们飞上三万英尺高空, 我记得飞机平稳降落。 我记得来接机的当地人给我带上的花环和插在耳鬓的鸡蛋花。 我记得...
    蔡沁潼阅读 1,819评论 4 4

友情链接更多精彩内容