list-comprehension引发RuntimeError

如何解决list-comprehension引发RuntimeError

为什么这段代码能很好地工作并且不会引发异常？

def myzip(*args):
    iters = [iter(arg) for arg in args]
    try:
        while True:
            yield tuple([next(it) for it in iters])
    except stopiteration:
        return


for x,y,z in myzip([1,2],[3,4],[5,6]):
    print(x,z)

但是如果这行

yield tuple([next(it) for it in iters])

替换为

yield tuple(next(it) for it in iters)

然后一切都停止工作并抛出RuntimeError？

解决方法

这是Python 3.5中引入的功能，而不是错误。对于PEP-479，当从生成器内部引发RuntimeError时，有意地重新引发StopIteration，这样，仅当生成器返回时，才能停止基于生成器的迭代，此时指向StopIteration异常会停止迭代。

否则，在Python 3.5之前，在生成器中任何地方引发的StopIteration异常将停止生成器，而不是传播它，因此在以下情况下：

a = list(F(x) for x in xs)
a = [F(x) for x in xs]

如果F(x)在迭代过程中的某个时刻引发StopIteration异常，前者将被截断，这使得调试变得困难，而后者将传播{{1 }}。该功能的目的是使两个语句的行为相同，这就是为什么更改会影响生成器但不影响列表理解的原因。