分块生成器[重复]

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