itertools拼装迭代器
连接多个迭代器
内置的itertools模块有一些函数可以把多个迭代器连城一个使用。
chain
chain可以把多个迭代器从头到尾连成一个迭代器。
python
import itertools
it = itertools.chain([1, 2, 3], [4, 5, 6])
print(list(it))
>>>
[1, 2, 3, 4, 5, 6]repeat
Repeat可以制作这样一个迭代器，它会不停地输出某个值。调用repeat时，也可以通过第二个参数指定迭代器最多能输出几次。
python
it = itertools.repeat('hello', 3)
print(list(it))
>>>
['hello', 'hello', 'hello']cycle
cycle可以制作这样一个迭代器，它会循环地输出某段内容之中的各项元素。
python
it = itertools.cycle([1, 2])
result = [next(it) for _ in range(10)]
print(result)
>>>
[1, 2, 1, 2, 1, 2, 1, 2, 1, 2]tee
tee可以让一个迭代器分裂成多个平行的迭代器，具体个数由第二个参数指定。如果这些迭代器推进的速度不一致，那么程序可能要用大量内存做缓冲，以存放进度落后的迭代器将来会用到的元素。
python
import itertools
it1, it2, it3 = itertools.tee(['first', 'second'], 3)
print(list(it1))
print(list(it2))
print(list(it3))
>>>
['first', 'second']
['first', 'second']
['first', 'second']zip_longest
它与Python内置的zip函数类似，但区别在于，如果源迭代器的长度不同，那么它会用fillvalue参数的值来填补提前耗尽的那些迭代器所留下的空缺。
python
import itertools
keys = ['one', 'two', 'three', 'four', 'five']
values = [1, 2]
normal = list(zip(keys, values))
print('zip: ', normal)
it = itertools.zip_longest(keys, values, fillvalue='nope')
longest = list(it)
print('zip_longest: ', longest)
>>>
zip: [('one', 1), ('two', 2)]
zip_longest: [('one', 1), ('two', 2), ('three', 'nope'), ('four', 'nope'), ('five', 'nope')]