Python的sum、map、filter和reduce

最近在看《Think Python》(英文版),看到了讲解map, reduce, filter等函数,觉得讲解的思路特别好。所以,我加上了自己的理解,写了本篇文章。

引子

如果要对列表中的数字求和,我们可以这样做:

def add_all(t):
    """t is a list of nums"""
    total = 0
    for x in t:
        total += x

    return total

运行结果:

《Python的sum、map、filter和reduce》 运行结果

由于python中求和操作太常见了,所以python提供了内置函数sum来进行这项操作

sum

将上一版代码修改如下:

if __name__ == '__main__':
    t = [1, 2, 3, 4]
    print sum(t)

运行结果:

《Python的sum、map、filter和reduce》 运行结果

像这种从一个列表中得出一个结果的操作,也叫reduce,除了求和这一种reduce操作外,python还支持自定义的方式

reduce

比如,我不是想对队列求和,而是想得到列表各数字的乘积,可以利用reduce这样实现:

def multiple_all(x, y):
    return x * y

if __name__ == '__main__':
    t = [1, 2, 3, 4]
    print reduce(multiple_all, t)

运行结果:

《Python的sum、map、filter和reduce》 运行结果

上面介绍了将一个从一个列表得到一个结果的情形,还有一种情形是:按照某个规律对列表中的元素一一转换,这就要用到map内置函数了

map

如果给定一个列表(元素为字符串),要把列表元素首字母大写,可以这样做

if __name__ == '__main__':
    t = ['hello', 'world', 'yarving']
    print map(lambda x: x.capitalize(), t)

运行结果:

《Python的sum、map、filter和reduce》 运行结果

还有一种情况,是要将列表里的元素过滤出去,可以用到filter函数

filter

给定一个列表(元素为数字),如果要仅保留不大于4的数字,可以这样做:

if __name__ == '__main__':
    t = [1, 2, 3, 4, 5, 6, 7]
    print filter(lambda x: x <= 4, t)

运行结果:

《Python的sum、map、filter和reduce》 运行结果

获取《Think Python》(英文版)电子书

如想要获取《Think Python》(英文书)的PDF版本,可发送邮件到 yarving@qq.com ,并标明主题 “Think Python” 。

    原文作者:yarving
    原文地址: https://www.jianshu.com/p/8e1961bbbe65
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞