Python 等效于 c++ find_if

Question

是否有一个与std::find_if 等效的内置python 来查找给定条件为true 的列表的第一个元素？换句话说，类似于列表的index() 函数，但具有任意一元谓词，而不仅仅是相等性测试。

我不想使用列表推导，因为我想到的特定谓词计算起来有些昂贵。

Answer 1

使用an answer to a related question 的提示，并借用 taras 发布的答案，我想出了这个：

>>> lst=[1,2,10,3,5,3,4]
>>> next(n for n in lst if n%5==0)
10

稍作修改会给你索引而不是值：

>>> next(idx for idx,n in enumerate(lst) if n%5==0)
2

现在，如果没有匹配，这将引发异常 StopIteration。如果没有匹配项，您可能需要使用处理异常并返回 None 的函数：

def first_match(iterable, predicate):
    try:
        return next(idx for idx,n in enumerate(iterable) if predicate(n))
    except StopIteration:
        return None

lst=[1,2,10,3,5,3,4]
print(first_match(lst, lambda x: x%5 == 0))

请注意，这使用生成器表达式，而不是列表推导式。列表推导会将条件应用于列表的每个成员并生成所有匹配项的列表。这会将其应用于每个成员，直到找到匹配项然后停止，这是解决问题的最少工作量。

Answer 2

比如说，你有一些谓词函数pred 和一个列表lst。 您可以使用itertools.dropwhile 获取lst 中的第一个元素， 其中pred 返回True 和

itertools.dropwhile(lambda x: not pred(x), lst).next()

它会跳过所有 pred(x) 为 False 和 .next() 的元素 为您生成 pred(x) 为 True 的值。

编辑：

用于查找lst 中的第一个元素可被5 整除的示例

>>> import itertools
>>> lst = [1,2,10,3,5,3,4]
>>> pred = lambda x: x % 5 == 0 
>>> itertools.dropwhile(lambda x: not pred(x), lst).next()
10