Python的sum、map、filter和reduce

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

引子

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

1
2
3
4
5
6
7
def add_all(t):
"""t is a list of nums"""
total = 0
for x in t:
total += x

return total

运行结果:
运行结果

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

sum

将上一版代码修改如下:

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

运行结果:
运行结果

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

reduce

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

1
2
3
4
5
6
def multiple_all(x, y):
return x * y

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

运行结果:
运行结果

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

map

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

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

运行结果:
运行结果

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

filter

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

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

运行结果:
运行结果

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

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

0%