分块生成器

Question

我在这里做错了什么。尝试使用 islice 获取生成器的块，但似乎是运行无限。

from itertools import islice

size = 2

def g():
    for x in range(11):
        print("generating: ", x)
        yield x

while True:
    chunk = islice(g(), size)
    if not chunk:
        break
    print("at chunk")
    for c in chunk:
        print(c)

我得到一个输出，它似乎永远循环并且似乎没有递增：

at chunk
generating:  0
0
generating:  1
1
at chunk
generating:  0
0
generating:  1
1
at chunk
generating:  0
0
generating:  1
1
at chunk

Answer 1

每次调用 g() 都会从头重新启动生成器。您需要将结果分配给一个变量，以便它继续使用相同的生成器。

并且如评论中所述，islice 对象始终为真。要判断您是否到达终点，请检查 for c in chunk: 循环是否做了任何事情。

from itertools import islice

def g():
    for x in range(11):
        print("generating: ", x)
        yield x

size = 2
gen = g()
while True:
    chunk = islice(gen, size)

    print("at chunk")
    empty = True
    for c in chunk:
        print(c)
        empty = False

    if empty:
        break

分块生成器

Chunking a generator

python

generator

itertools