如何让 Python 生成器返回 None 而不是 StopIteration？

Question

我正在使用生成器在列表中执行搜索，就像这个简单的示例一样：（

>>> a = [1,2,3,4]
>>> (i for i, v in enumerate(a) if v == 4).next()
3

只是为了稍微说明一下示例，与上面的列表相比，我使用的列表要长得多，并且条目比 int 稍微复杂一点 这样做是为了每次搜索时都不会遍历整个列表）

现在，如果我将其更改为 i == 666，它将返回 <代码>StopIteration 因为它在 a 中找不到任何 666 条目。

我怎样才能让它返回 None 呢？我当然可以将它包装在 try ... except 子句中，但是有没有更 Pythonic 的方法来做到这一点？

Answer 1

如果您使用的是 Python 2.6+，则应该使用 next 内置函数函数，而不是 next 方法（在 3.x 中被 __next__ 替换）。如果迭代器耗尽，next 内置函数会采用一个可选的默认参数来返回，而不是引发 StopIteration：

next((i for i, v in enumerate(a) if i == 666), None)

Answer 2

您可以使用 (None,) 链接生成器：

from itertools import chain
a = [1,2,3,4]
print chain((i for i, v in enumerate(a) if v == 6), (None,)).next()

但我认为 a.index(2) 不会遍历完整列表，当找到 2 时，搜索完成。你可以测试一下：

>>> timeit.timeit("a.index(0)", "a=range(10)")
0.19335955439601094
>>> timeit.timeit("a.index(99)", "a=range(100)")
2.1938486138533335