为什么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异常被抛出在生成器表达式中。它只会终止生成器表达式，而不是封闭的生成器函数。