在使用 map(next, iterables) 的 StopIteration 之前引发 ValueError

Question

我正在编写这样的 4 个列表迭代器的循环：

it0 = iter(['foo','bar'])
it1 = iter(['bar','foo'])
it2 = iter(['foo','foo'])
it3 = iter(['bar','bar'])

try:
    a, b, c, d = map(next, (it0, it1, it2, it3))
    while True:
        #some operations that can raise ValueError
        if a+b+c+d == 'foobar'*2:
            a, b, c, d = map(next, (it0, it1, it2, it3))
        elif a+b == 'foobar':
            c, d = map(next,(it2, it3))
        else:
            a, b = map(next,(it0, it1))
except StopIteration:
    pass

但是当我在 Python3 中运行这段代码时，我得到了一个 ValueError: not enough values to unpack (expected 2, got 0)，在预期的 StopIteration 之前提出。

我不想在except 语句中捕获ValueError，因为while 循环中的某些操作也可能导致应该停止程序的ValueError。

next 怎么在赋值前没有抛出异常？

在 Python2.7 中，首先引发 StopIteration。

Answer 1

StopIteration 被提升，但元组赋值将其吞下，因为它将 StopIteration 视为来自 map() 迭代器的信号，表明 它 已完成生成值：

>>> i0, i1 = iter(['foo']), iter([])
>>> m = map(next, (i0, i1))
>>> next(m)
'foo'
>>> next(m)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>> i0, i1 = iter(['foo']), iter([])
>>> m = map(next, (i0, i1))
>>> a, b = m
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: not enough values to unpack (expected 2, got 1)

这是正常行为；期望迭代器作为输入的 Python 内置函数始终使用 StopIteration 作为“迭代完成”信号，并且元组解包必须在此处迭代。

要么先将map()输出转换为列表并测试长度，在每个可迭代对象上单独使用next()而不使用map()，或者在本地捕获ValueError。

测试长度必须重新提出StopIteration：

values = list(map(next, (it0, it1, it2, it3)))
if len(values) < 4:
    raise StopIteration
a, b, c, d = values

注意这里list() 已经吞下了StopIteration 异常。

捕获ValueError 只是为了map() 操作：

try:    
    a, b, c, d = map(next, (it0, it1, it2, it3))
except ValueError:
    raise StopIteration

在每个迭代器上单独调用next()，根本不使用map()：

a, b, c, d = next(it0), next(it1), next(it2), next(it3)

或使用列表推导：

a, b, c, d = [next(i) for i in (it0, it1, it2, it3)]

两者都确保在分配发生之前调用next()，而不是在分配本身期间。

Answer 2

根据 Martijn Pieters 的回答，可以直接使用列表推导：

a, b, c, d = [next(it) for it in (it0, it1, it2, it3)]

这个post 阐明了为什么for 在可迭代但不在循环体中捕获StopIteration。

另一种可能的方式，利用next 的default 参数：

it0 = iter(['foo','bar'])
it1 = iter(['bar','foo'])
it2 = iter(['foo','foo'])
it3 = iter(['bar','bar'])

try:
    a, b, c, d = map(next, (it0, it1, it2, it3), (None,)*4)
    while all(_ is not None for _ in (a,b,c,d)):
        #some operations that can raise ValueError
        if a+b+c+d == 'foobar'*2:
            a, b, c, d = map(next, (it0, it1, it2, it3), (None,)*4)
        elif a+b == 'foobar':
            c, d = map(next,(it2, it3), (None,)*2)
        else:
            a, b = map(next,(it0, it1), (None,)*2)
except StopIteration:
    pass