重置 itertools.count

Question

我想在某种情况发生时重置计数。我尝试执行以下操作但没有成功！

c = itertools.count(1)
for i in range(100):
    next(c)
    print(c)
    if next(c) == 10:
    c = itertools.count(1)

Answer 1

您应该使用 next returns 进行比较，而不是 itertools.count 本身的实例：

import itertools
c = itertools.count(1)
for i in range(100):
    value = next(c)
    print(c)
    if value == 10:
        c = itertools.count(1)

请注意，以下代码不会重置现有的 itertools.count 实例，而是在满足条件时将 itertools.count 的全新实例置于 c 之下。