itertools模块

Posted 2020-11-02 瓜田月夜

Python的内建模块itertools提供了非常有用的用于操作迭代对象的函数。

无限迭代器

count()会创建一个无限迭代器

import itertools
natuals = itertools.count(1)
for n in natuals:
    print(n)
# 打印根本停不下来

cycle()会把传入的一个序列无限重复下去：

import itertools
natuals = itertools.cycle(‘ABC‘)
for n in natuals:
    print(n)
# 也是根本停不下来

repeat()负责把一个元素无限的重复下去,不过如果提供第二个参数就可以限定重复次数:

import itertools
natuals = itertools.repeat(‘A‘, 10)
for n in natuals:
    print(n)
# 打印了十次A

无限序列只有在for迭代时才会无限地迭代下去，如果只是创建了一个迭代对象，它不会事先把无限个元素生成出来，事实上也不可能在内存中创建无限多个元素。

无限序列虽然可以无限迭代下去，但是通常我们会通过takewhile()等函数根据条件判断来截取出一个有限的序列：

import itertools
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals)
# takewhile接受函数和一个可迭代对象,会将迭代对象的值传入函数,返回False则停止迭代
for n in ns:
    print(n)

比较有用的几个函数

chain()可以把以组迭代对象串联起来,形成一个更大的迭代器

for c in itertools.chain(‘ABC‘, ‘XYZ‘):
    print(c)
# 迭代效果：‘A‘ ‘B‘ ‘C‘ ‘X‘ ‘Y‘ ‘Z‘

groupby()把迭代器中相邻的重复元素跳出来放在一起