python - python中相同生成器的链式重复

Question

考虑以下简单的生成器:

def simple_gen():
    for number in range(5):
        yield number ** 2

我愿意使用 itertools.repeat 和 itertools.chain 将生成器链接到自身 n 次。为清楚起见，请考虑以下相同的(非生成器)示例:

array = [1,2,3,4]
repetitions = itertools.repeat( array ,2)
list(itertools.chain.from_iterable(repetitions)) -> [1,2,3,4,1,2,3,4]

我想要相同的但使用我自己的生成器 (simple_gen) 而不是数组。当然，简单的替换是行不通的，因为 itertools.repeat 重复相同的对象，因此生成器的后续重复将被耗尽。

关于如何使用 itertools 模块实现此目的的一些想法？

我不想将生成器转换为列表或其他容器。

Answer 1

您可以先将生成器输出转换为列表:

repeat(list(simple_gen()), 2)

否则您不能重复生成器输出。您最多可以重新创建生成器count 次:

from itertools import repeat

def recreate(callable, count=None):
    for c in repeat(callable, count):
        yield from c()

在 Python 3 和

from itertools import repeat

def recreate(callable, count=None):
    for c in repeat(callable, count):
        for val in c():
            yield val

对于 Python 2 并使用它代替 chain.from_iterable(repeat(callable(), count))。请注意，生成器函数未被调用，而是传入函数对象本身。

演示:

>>> from itertools import repeat
>>> def simple_gen():
...     for number in range(5):
...         yield number ** 2
...
>>> def recreate(callable, count=None):
...     for c in repeat(callable, count):
...         for val in c():
...             yield val
...
>>> list(recreate(simple_gen, 2))
[0, 1, 4, 9, 16, 0, 1, 4, 9, 16]