python中itertools模块使用小结

Python中itertools模块使用小结

Python中itertools是一个标准库，用于生成迭代器的函数和无限迭代器。它提供了各种有用的迭代器用于有效地对迭代器工作。下面是一些最常用的itertools函数：

itertools.count(start=0, step=1)

生成从start开始的连续整数，步骤为step。

import itertools

for i in itertools.count(10,2):
    print(i)
    if i > 20:
        break

以上代码输出如下：

10
12
14
16
18
20
22

itertools.cycle(iterable)

循环遍历给定迭代器中的所有项。

import itertools

x = 0

for i in itertools.cycle(['a', 'b', 'c']):
    print(i)
    x += 1
    if x > 7:
        break

以上代码输出如下：

a
b
c
a
b
c
a
b

除此之外，itertools还有许多其他有用的函数，如itertools.permutations和itertools.combinations等等，但以上两个函数是最为常用的itertools函数。