分块发电机

问题描述

我在这里做错了什么。尝试使用 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

解决方法

每次调用 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