如何解决分块发电机
我在这里做错了什么。尝试使用 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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。