python - 如何在 Python 中显式检测生成器端？

Question

我写了一个生成器，它返回值的元组:

import numpy as np

def mygenerator():
    data = np.arange(10)
    data = np.reshape(data, (-1, 2))
    for row in data:
        yield row[0], row[1]

generator = mygenerator()
while True:
    a, b = generator.__next__()
    print("a: ", a, "b: ", b)

并且想在没有 for ... in 的情况下使用它.那么如何检测世代结束呢？

Answer 1

生成器会抛出异常 StopIteration当它耗尽时。您可以使用 try 捕获异常… except语句和退出循环发生时:

while True:
    try:
        a, b = next(generator)
    except StopIteration:
        break
    print("a: ", a, "b: ", b)

顺便说一句，为什么不 for环形？看起来:

for a, b in generator:
    print("a: ", a, "b: ", b)

完全符合您的要求，没有繁琐的结构。