python - 返回列表中匹配条件的第一项

Question

这个问题在这里已经有了答案:

Find first sequence item that matches a criterion [duplicate] (2 个回答)

关闭 9年前。

我有一个功能。 matchCondition(a)，它接受一个整数并返回 True 或 False。

我有一个包含 10 个整数的列表。我想返回列表中的第一项(与原始列表的顺序相同)，其中 matchCondition 返回 True。

尽可能像 Python 一样。

Answer 1

next(x for x in lst if matchCondition(x)) 

应该可以，但是如果列表中没有任何元素匹配，它将引发 StopIteration。您可以通过向 next 提供第二个参数来抑制它:

next((x for x in lst if matchCondition(x)), None)

如果没有匹配，将返回 None。

演示:

>>> next(x for x in range(10) if x == 7)  #This is a silly way to write 7 ...
7
>>> next(x for x in range(10) if x == 11)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>> next((x for x in range(10) if x == 7), None)
7
>>> print next((x for x in range(10) if x == 11), None)
None

---

最后，为了完整起见，如果你想要 all 列表中匹配的项目，这就是内置 filter 函数的用途:

all_matching = filter(matchCondition,lst)

在 python2.x 中，这返回一个列表，但在 python3.x 中，它返回一个可迭代对象。