Python itertools accumulate函数详解

一.定义

itertools.accumulate(iterable[, func])

二.解析

iterable是一个可迭代的对象，如list等。

accumulate函数的功能是对传进来的iterable对象逐个进行某个操作（默认是累加，如果传了某个fun就是应用此fun

比如iterable=[1,2,3,4] 默认会先累加iterable 0～0（1）， 然后0～1（1+2），最后0～3（1+2+3）

结果会是[1,3,6,10]

注意：accumulate函数返回是一个可迭代对象，可以用在for里面，而不是最后的累加结果，如果我们想要的是直接的结果

需要强制转化类型，比如转化成list，详细可以给下面的示例

三.代码示例

>>> a=[1,2,3,4,5]
>>> b=accumulate(a)  #默认是累加
>>> b   #这里返回的是一个可迭代对象
<itertools.accumulate object at 0x7f3e5c2f4e48>
>>> list(b)   #强制转化
[1, 3, 6, 10, 15]
>>> 

传递自己想要的操作fun，比如换成乘法

>>> import operator
>>> c=accumulate(a,operator.mul)
>>> c
<itertools.accumulate object at 0x7f3e5c2f4f88>
>>> list(c)
[1, 2, 6, 24, 120]
>>>