Python的itertools包的常用方法總結

import itertools
for i in itertools.chain([1,2],[3,4]):#拼接列表並迭代
print(i)
for i in itertools.chain((1,2),{
3:1,'a':3}):#列表、元組、集合、字典可以任意組合
print(i)
sum=0
for i in itertools.cycle([1,2]):#無限循環迭代，但是這裡只能傳一個參數
print(i)
if sum>2:
break
sum+=1
for i in itertools.accumulate([1,2,3]):#計算累加和，這裡也只能傳一個參數
print(i)
for i in itertools.accumulate(['a','b','c']):#計算累加和，字符串也可以拼接
print(i)
for i in itertools.accumulate([1,2,3],lambda a,b:a*b):#可以傳入自定義函數，這裡計算累積
print(i)