如何返回发电机输出

Question

我想弄清楚如何返回生成器输出。我知道next(generator)返回函数的输出，但是我想知道如何返回多个输出。

例如：

alist = [1,2,3,4,5,6]
def aiterator():
    for i in alist:
        yield i+1

generator = aiterator()

我知道next(generator)将返回1，但是如何返回2,3，或者如果我想返回2,3,4，而又没有键入next(generator)两次或三次？

具体地说，我正在考虑另一个函数，该函数将返回next(generator)的数量：

for i in aiterator():
  return ?

Answer 1

如果要一一输出整个列表，请使用：

while True:
  next(generator)

如果您想要更多的自定义输出，则如注释之一所述，更改yield值。