python reduce 函数使用详解


Posted in Python onDecember 05, 2017

reduce() 函数在 python 2 是内置函数, 从python 3 开始移到了 functools 模块。

官方文档是这样介绍的

reduce(...)
reduce(function, sequence[, initial]) -> value

Apply a function of two arguments cumulatively to the items of a sequence,
from left to right, so as to reduce the sequence to a single value.
For example, reduce(lambda x, y: x+y, [1, 2, 3, 4, 5]) calculates
((((1+2)+3)+4)+5). If initial is present, it is placed before the items
of the sequence in the calculation, and serves as a default when the
sequence is empty.

从左到右对一个序列的项累计地应用有两个参数的函数,以此合并序列到一个单一值。

例如,reduce(lambda x, y: x+y, [1, 2, 3, 4, 5])  计算的就是((((1+2)+3)+4)+5)。

如果提供了 initial 参数,计算时它将被放在序列的所有项前面,如果序列是空的,它也就是计算的默认结果值了

嗯, 这个文档其实不好理解。看了还是不懂。 序列 其实就是python中 tuple  list  dictionary string  以及其他可迭代物,别的编程语言可能有数组。

reduce 有 三个参数

function 有两个参数的函数, 必需参数
sequence tuple ,list ,dictionary, string等可迭代物,必需参数
initial 初始值, 可选参数

reduce的工作过程是 :在迭代sequence(tuple ,list ,dictionary, string等可迭代物)的过程中,首先把 前两个元素传给 函数参数,函数加工后,然后把得到的结果和第三个元素作为两个参数传给函数参数, 函数加工后得到的结果又和第四个元素作为两个参数传给函数参数,依次类推。 如果传入了 initial 值, 那么首先传的就不是 sequence 的第一个和第二个元素,而是 initial值和 第一个元素。经过这样的累计计算之后合并序列到一个单一返回值

 reduce 代码举例,使用REPL演示

>>> def add(x, y):
...   return x+y
...
>>> from functools import reduce
>>> reduce(add, [1,2,3,4])
10
>>>

上面这段 reduce 代码,其实就相当于 1 + 2 + 3 + 4 = 10, 如果把加号改成乘号, 就成了阶乘了
当然 仅仅是求和的话还有更简单的方法,如下

>>> sum([1,2,3,4])
10
>>>

很多教程只讲了一个加法求和,太简单了,对新手加深理解还不够。下面讲点更深入的例子

还可以把一个整数列表拼成整数,如下

>>> from functools import reduce
>>> reduce(lambda x, y: x * 10 + y, [1 , 2, 3, 4, 5])
12345
>>>

对一个复杂的sequence使用reduce ,看下面代码,更多的代码不再使用REPL, 使用编辑器编写

from functools import reduce
 scientists =({'name':'Alan Turing', 'age':105},
       {'name':'Dennis Ritchie', 'age':76},
       {'name':'John von Neumann', 'age':114},
       {'name':'Guido van Rossum', 'age':61})
 def reducer(accumulator , value):
   sum = accumulator['age'] + value['age']
   return sum
 total_age = reduce(reducer, scientists)
 print(total_age)

这段代码会出错,看下图的执行过程

python reduce 函数使用详解

所以代码需要修改

from functools import reduce
 scientists =({'name':'Alan Turing', 'age':105, 'gender':'male'},
       {'name':'Dennis Ritchie', 'age':76, 'gender':'male'},
       {'name':'Ada Lovelace', 'age':202, 'gender':'female'},
       {'name':'Frances E. Allen', 'age':84, 'gender':'female'})
 def reducer(accumulator , value):
   sum = accumulator + value['age']
   return sum
 total_age = reduce(reducer, scientists, 0)
 print(total_age)

7, 9 行 红色部分就是修改 部分。 通过 help(reduce) 查看 文档,
reduce 有三个参数, 第三个参数是初始值的意思,是可有可无的参数。

修改之后就不出错了,流程如下

python reduce 函数使用详解

这个仍然也可以用 sum 来更简单的完成

sum([x['age'] for x in scientists ])

做点更高级的事情,按性别分组

from functools import reduce
scientists =({'name':'Alan Turing', 'age':105, 'gender':'male'},
       {'name':'Dennis Ritchie', 'age':76, 'gender':'male'},
       {'name':'Ada Lovelace', 'age':202, 'gender':'female'},
       {'name':'Frances E. Allen', 'age':84, 'gender':'female'})
def group_by_gender(accumulator , value):
  accumulator[value['gender']].append(value['name'])
  return accumulator
grouped = reduce(group_by_gender, scientists, {'male':[], 'female':[]})
print(grouped)

输出

{'male': ['Alan Turing', 'Dennis Ritchie'], 'female': ['Ada Lovelace', 'Frances E. Allen']}

可以看到,在 reduce 的初始值参数传入了一个dictionary,, 但是这样写 key 可能出错,还能再进一步自动化,运行时动态插入key

修改代码如下

grouped = reduce(group_by_gender, scientists, collections.defaultdict(list))

当然 先要 import  collections 模块

这当然也能用 pythonic way 去解决

import itertools
scientists =({'name':'Alan Turing', 'age':105, 'gender':'male'},
       {'name':'Dennis Ritchie', 'age':76, 'gender':'male'},
       {'name':'Ada Lovelace', 'age':202, 'gender':'female'},
       {'name':'Frances E. Allen', 'age':84, 'gender':'female'})
grouped = {item[0]:list(item[1])
      for item in itertools.groupby(scientists, lambda x: x['gender'])}
print(grouped)

再来一个更晦涩难懂的玩法。工作中要与其他人协作的话,不建议这么用,与上面的例子做同样的事,看不懂无所谓。

from functools import reduce
scientists =({'name':'Alan Turing', 'age':105, 'gender':'male'},
       {'name':'Dennis Ritchie', 'age':76, 'gender':'male'},
       {'name':'Ada Lovelace', 'age':202, 'gender':'female'},
       {'name':'Frances E. Allen', 'age':84, 'gender':'female'})
grouped = reduce(lambda acc, val: {**acc, **{val['gender']: acc[val['gender']]+ [val['name']]}}, scientists, {'male':[], 'female':[]})
print(grouped)

**acc, **{val['gneder']...   这里使用了 dictionary merge syntax ,  从 python 3.5 开始引入, 详情请看 PEP 448 - Additional Unpacking Generalizations  怎么使用可以参考这个 python - How to merge two dictionaries in a single expression? - Stack Overflow

python 社区推荐写可读性好的代码,有更好的选择时不建议用reduce,所以 python 2 中内置的reduce 函数 移到了 functools模块中

Python 相关文章推荐
python删除文件示例分享
Jan 28 Python
Python MySQLdb 使用utf-8 编码插入中文数据问题
Mar 13 Python
python实现输入数字的连续加减方法
Jun 22 Python
Python3 利用requests 库进行post携带账号密码请求数据的方法
Oct 26 Python
python如何制作英文字典
Jun 25 Python
python用for循环求和的方法总结
Jul 08 Python
使用python实现希尔、计数、基数基础排序的代码
Dec 25 Python
使用python处理题库表格并转化为word形式的实现
Apr 14 Python
如何使用python切换hosts文件
Apr 29 Python
Python 实现简单的客户端认证
Jul 29 Python
简单了解python关键字global nonlocal区别
Sep 21 Python
Python Pygame实现俄罗斯方块
Feb 19 Python
有趣的python小程序分享
Dec 05 #Python
详细分析python3的reduce函数
Dec 05 #Python
Python数据可视化正态分布简单分析及实现代码
Dec 04 #Python
Python编程实现二分法和牛顿迭代法求平方根代码
Dec 04 #Python
Python编程给numpy矩阵添加一列方法示例
Dec 04 #Python
Python实现返回数组中第i小元素的方法示例
Dec 04 #Python
Python实现基本数据结构中队列的操作方法示例
Dec 04 #Python
You might like
php知道与问问的采集插件代码
2010/10/12 PHP
解析file_get_contents模仿浏览器头(user_agent)获取数据
2013/06/27 PHP
Phpstorm+Xdebug断点调试PHP的方法
2018/05/14 PHP
jQuery使用手册之二 DOM操作
2007/03/24 Javascript
JAVASCRIPT下判断IE与FF的比较简单的方式
2008/10/17 Javascript
在Iframe中获取父窗口中表单的值(示例代码)
2013/11/22 Javascript
javascript性能优化之事件委托实例详解
2015/12/12 Javascript
JS组件系列之Bootstrap Icon图标选择组件
2016/01/28 Javascript
利用jQuery的动画函数animate实现豌豆发射效果
2016/08/28 Javascript
基于JavaScript实现随机颜色输入框
2016/12/10 Javascript
es6在react中的应用代码解析
2017/11/08 Javascript
vue axios 在页面切换时中断请求方法 ajax
2018/03/05 Javascript
详解如何在vue-cli中使用vuex
2018/08/07 Javascript
解决vue select当前value没有更新到vue对象属性的问题
2018/08/30 Javascript
JavaScript数组去重的方法总结【12种方法,号称史上最全】
2019/02/28 Javascript
[02:36]DOTA2英雄基础教程 斯拉克
2013/11/29 DOTA
Python编程实现双击更新所有已安装python模块的方法
2017/06/05 Python
python3 判断列表是一个空列表的方法
2018/05/04 Python
python中使用 xlwt 操作excel的常见方法与问题
2019/01/13 Python
python利用Opencv实现人脸识别功能
2019/04/25 Python
Python全栈之列表数据类型详解
2019/10/01 Python
python带参数打包exe及调用方式
2019/12/21 Python
如何在Anaconda中打开python自带idle
2020/09/21 Python
TripAdvisor越南:全球领先的旅游网站
2017/09/21 全球购物
戴尔荷兰官方网站:Dell荷兰
2020/10/04 全球购物
香港艺人陈冠希创办的潮流品牌:JUICESTORE
2021/03/04 全球购物
CSS代码检查工具stylelint的使用方法详解
2021/03/27 HTML / CSS
夜大毕业生自我评价分享
2013/11/10 职场文书
《那片绿绿的爬山虎》教学反思
2014/02/27 职场文书
就职演讲稿范文
2014/05/19 职场文书
元旦联欢会策划方案
2014/06/11 职场文书
大学班长竞选稿
2015/11/20 职场文书
2016秋季幼儿园开学寄语
2015/12/03 职场文书
《蓝鲸的眼睛》读后感5篇
2020/01/15 职场文书
golang中的空slice案例
2021/04/27 Golang
JavaScript实现两个数组的交集
2022/03/25 Javascript