python - 计数到无穷大的生成器

Question

这个问题已经有答案了:

Is there an expression for an infinite iterator? (7 个回答)

已关闭 7 年前。

只是为了好玩和好奇，我设计了这种编写一个计数为无穷大的生成器的方法，至少在原则上是这样。没有什么非常 python 式的。只是玩玩而已。

但显然每次迭代 lis变得更长，直到某些东西最终破裂。

还有其他(更好或更差)的方法来实现相同的目标吗？请分享。

lis = [1]
gen = ((k, lis.append(k+1))[0] for k in lis)

使用它

for j in gen:
    print(j)

Answer 1

您可以使用itertools.count :

import itertools

for i in itertools.count(1):
    print(i)

从上面的文档来看，这在功能上相当于:

def count(start=0, step=1):
    # count(10) --> 10 11 12 13 14 ...
    # count(2.5, 0.5) -> 2.5 3.0 3.5 ...
    n = start
    while True:
        yield n
        n += step

你可以像这样使用它:

>>> my_generator = count(3, 2)
>>> next(my_generator)
3
>>> next(my_generator)
5
>>> next(my_generator)
7