如何用itertools生成无限序列

Question

我想生成两个0和1的不定式系列，特别是按以下顺序生成：

0、1、0，-1、0、1、0，-1，...

我创建了以下代码，该代码未返回除了：

# for in loop
for i in itertools.cycle(range(0,2)):
    if i == 0:
        i += 1
        if i == 1:
            i -= 1
            if i == 0:
                i -= 1
                print(i, end = " ")

它只是返回一系列-1。无法弄清楚错误在哪里。任何人都可以给任何建议吗

Answer 1

您可以使用 itertools.cycles.cycle（） 以明确说明要生成的元素的方式：

from itertools import cycle
cycle([0, 1, 0, -1])

Answer 2

作为替代方案，您可以使用一些简单的模块化算术来实现自己的发电机：

def seq():
    i = 0
    while True:
        yield (-1)**(i // 2) * (i % 2)
        i = (i + 1) % 4  # keeps i small, so as not to take up too much memory

演示：

>>> s = seq()
>>> [next(s) for _ in range(10)]
[0, 1, 0, -1, 0, 1, 0, -1, 0, 1]

我知道OP指出，他们特别想使用Itertools，但这仍然可能对他人有帮助。

正如评论中指出的那样，您也可以简单地模拟itertools.cycle通过传递任意args并使用的收益率（请注意itertools.cycles.cycle实际上，在引擎盖下的工作方式不同）：

def seq(*args):
    while True:
        yield from args


s = seq(0, 1, 0, -1)

但是，在我看来，这并不像弄清楚实际算术序列那样有趣：D

Answer 3

不确定为什么您要坚持使用itertools.cycle在这样的循环中，但这是您可以做这项工作的一种方法：

for i in itertools.cycle(range(0, 2)):
    if i == 0:
        print(i, end=" ")
        i += 1
        print(i, end=" ")
        i -= 1
        print(i, end=" ")
        i -= 1
        print(i, end=" ")

请注意，确实没有意义使用嵌套如果语句，因为每个嵌套谓词将始终为真：如果i为零，然后您添加一个，则当然您的嵌套谓词如果i == 1是正确的...

此外，您对itertools.cycles.cycle在 for 中的使用是一个反模式。如BrokenBenchmark的答案所示，您可以永远循环序列本身的元素。如果您坚持使用对循环使用：

for i in itertools.cycle([0, 1, 0, -1]):
    print(i, end=" ")

但是，这种方法有很多错误。也就是说，正如我在评论中所讨论的那样，这是一个无限 loop ，它与无限 序列 。有了一个无限的循环，再也不会发生任何其他事情 - 您的程序将永远被骑自行车在您的四个要素上骑自行车，无法做任何其他事情。

如果您使用itertools.cycle作为生成器，则可以随时在序列中关闭的位置，允许您执行其他任务按照您想要的序列，如您所需的那样：

c = itertools.cycle((0, 1, 0, -1))
next(c) # 0
next(c) # 1
next(c) # 0
next(c) # -1

# do something else for a while
next(c) # 0

# yield the next 10 items from the sequence
for _ in range(10):
    print(next(c), end=" ")

# do something else again