itertools

itertools是一个迭代工具模块,里面包括很多功能, 返回一个迭代器对象

itertools.chain

把多个序列组合在一起变成一个序列

>>> sequence = ((1, 2, 3, 4, 5, 6), (6, 5, 4, 3, 2, 1), [10, 20, 30, 40, 50, 60])
>>> itertools.chain(*sequence)                 # 返回一个迭代器
<itertools.chain object at 0x7fe0ca815c18>
>>> list(itertools.chain(*sequence))
[1, 2, 3, 4, 5, 6, 6, 5, 4, 3, 2, 1, 10, 20, 30, 40, 50, 60]
>>> 
>>> sum(itertools.chain(*sequence))           # 求和
252

itertools.ziplongest

zip_longest函数可以平行变量多个迭代器，而不用在乎它们的长度是否相等。有点像zip函数，但是zip里面的对象长度最好相等

>>> import itertools
>>> I = [1,1,2,2,3,4,4]
>>> L = [1,2,3,4,5,5,7,8]
>>> itertools.zip_longest(I, L)
<itertools.zip_longest object at 0x7f7c078fd7c8>
>>> list(itertools.zip_longest(I, L))
[(1, 1), (1, 2), (2, 3), (2, 4), (3, 5), (4, 5), (4, 7), (None, 8)]