博客

翻译自：

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