返回符合条件的列表中的第一项

Question

我有一个功能。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

最后，为了完整起见，如果您希望列表中 所有 匹配的项，那么内置filter函数就是为了：

all_matching = filter(matchCondition,lst)

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