翻译自:https://pymotw.com/3/itertools/index.html
itertools中函数设计的初衷是使用起来快速且更有效的利用内存,数据不会被创建直到真的需要,这种“lazy”模式使其不用存储大量数据在内存中。
组合和分割可迭代对象
1) chain
chain函数以多个iterators 为入参,返回一个iterators , 该iterators 包含了入参中的所有元素。
from itertools import *
for i in chain([1, 2, 3], ['a', 'b', 'c']): #使用chain的好处是可以不用再构建一个大的list
print(i, end=' ')
#结果
1 2 3 a b c
2) chain.from_iterable()函数
若iterables 事先不能确定,可以使用chain.from_iterable()函数
from itertools import *
def make_iterables_to_chain():
yield [1, 2, 3]
yield ['a', 'b', 'c']
for i in chain.from_iterable(make_iterables_to_chain()):
print(i, end=' ')
print()
#结果
1 2 3 a b c
3) zip函数
将多个iterator中的相应位置数值组成一个个tuple,注意对比zip(较短的iterator结束后就停止)和zip_longest函数(较长的iterator结束后才停止,默认用None填充没有的值)。
for i in zip([1, 2, 3], ['a', 'b', 'c']):
print(i)
#结果
(1, 'a')
(2, 'b')
(3, 'c')
4) islice函数
返回入参iterator中的根据索引选择的数值
from itertools import *
print('Stop at 5:')
for i in islice(range(100), 5):
print(i, end=' ')
print('\n')
print('Start at 5, Stop at 10:')
for i in islice(range(100), 5, 10):
print(i, end=' ')
print('\n')
print('By tens to 100:')
for i in islice(range(100), 0, 100, 10):
print(i, end=' ')
print('\n')
#结果
Stop at 5:
0 1 2 3 4
Start at 5, Stop at 10:
5 6 7 8 9
By tens to 100:
0 10 20 30