2012-04-11 59 views
8

为了更好地理解Python的发电机我试图实现itertools模块中的设施,并陷入困境与izip为什么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 

我的代码使用错误行,并参考实现(在手册中给出)使用OK线,不考虑其他微小差异。在这个片段:

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

我的代码输出:

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

,而预期输出是:

(1, 4) 
(2, 5) 

有什么错我的代码,好吗?

回答

12

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

+2

@wdscxsj:see http://www.velocityreviews.com/forums/t723967-generator-expressions-vs-comprehensions.html – 2012-04-11 15:19:12

+0

@Sven Marnach:谢谢!你的解释很简洁,非常重要。 – wdscxsj 2012-04-11 15:20:37

+0

@Steven Rumbalski:链接页面帮助我更好地理解。谢谢! – wdscxsj 2012-04-11 15:24:07