如何返回生成器输出[重复]

Question

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

例如：

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

generator = aiterator()

我知道next(generator) 会返回 1，但是如何在不输入 next(generator) 两次或三次的情况下返回 2,3，或者如果我愿意，返回 2,3,4？

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

for i in aiterator():
  return ?

Answer 1

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

while True:
  next(generator)

如果您想要更多自定义输出，则如其中一个 cmets 所述，更改 yield 值。