【Python笔记】 itertools模块（续更）

---

由于itertools是内置库，不需要任何安装，直接import itertools即可。

import itertools

1 accumulate

itertools.accumulate(iterable[, func ])

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

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

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

结果：[1,3,6,10]

注意：accumulate函数返回是一个可迭代对象，可以用在for里面
for sums in accumulate(arr):

需要强制转化类型，比如转化成list：

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

>>> list(b)   #强制转化
[1, 3, 6, 10, 15]

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

>>> import operator
>>> c=accumulate(a,operator.mul)
>>> c
<itertools.accumulate object at 0x7f3e5c2f4f88>

>>> list(c)
[1, 2, 6, 24, 120]