Python – Itertools.count()

itertools.count() 是 Python 标准库中的一个函数，它返回一个无限迭代器，该迭代器生成连续的整数，从指定的起始数开始。下面是 count() 函数的语法：

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

参数 start 和 step 都是可选的。start 参数指定了整数序列的起始数。step 参数指定了连续整数之间的步长，默认为 1。

下面是使用 itertools.count() 生成整数序列的例子：

import itertools

# 从 0 开始生成 10 个整数
for i in itertools.islice(itertools.count(), 10):
    print(i, end=' ')

# 从 1 开始，以 2 为步长，生成 10 个整数
for i in itertools.islice(itertools.count(1, 2), 10):
    print(i, end=' ')

上面代码的输出分别如下：

0 1 2 3 4 5 6 7 8 9
1 3 5 7 9 11 13 15 17 19

这里使用了 itertools.islice() 函数，它可以根据需要从迭代器中获取一定数量的元素。第一个参数是迭代器，第二个参数是要获取的元素数量。使用 itertools.islice() 可以避免无限迭代器导致的死循环。

使用 itertools.count() 可以很方便地生成整数序列，这在需要连续整数的场合非常有用。