重置 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返回的內容進行比較，而不是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下。