分块发电机

用户1179317

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

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。