为什么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其中一个迭代所耗尽的异常被抛出到生成器表达式中.它只会终止生成器表达式,而不是封闭的生成器函数.