python - 为什么 izip() 的这种实现不起作用？

Question

为了更好地理解 Python 的生成器，我正在尝试在itertools模块中实现设施，并遇到以下问题izip：

def izip(*iterables):
    its = tuple(iter(it) for it in iterables)
    while True:
        yield tuple(next(it) for it in its)  # ERROR
        # yield tuple(map(next, its))  # OK

我的代码使用 ERROR 行，参考实现（在手册中给出）使用 OK 行，不考虑其他微小差异。有了这个片段：

for x in izip([1, 2, 3], (4, 5)):
    print x

我的代码输出：

(1, 4)
(2, 5)
(3,)
()
()
...  # indefinite ()

，而预期的输出是：

(1, 4)
(2, 5)

请问我的代码有什么问题吗？

Answer 1

您的实现不起作用的原因是由于StopIteration一个迭代器被耗尽而导致的异常被抛出在生成器表达式中。它只会终止生成器表达式，而不是封闭的生成器函数。