Python - 函数式“查找”？

Question

我需要一个函数，它能够迭代集合，以集合的元素作为参数调用提供的函数，并在从提供的函数接收到“True”时返回参数或其索引。

它是这样的：

def find(f, seq, index_only=True, item_only=False):
     """Return first item in sequence where f(item) == True."""
     index = 0
     for item in seq:
         if f(item):
             if index_only:
                 return index
             if item_only:
                 return item
             return index, item
         index+= 1
     raise KeyError

所以我想知道标准Python工具集中是否有类似的东西？

Answer 1

尝试 itertools ，例如 ifilter。

Answer 2

我不认为有任何这样的函数具有如此精确的语义，无论如何你的函数很短，足够好，你可以轻松地改进它以供以后使用，所以使用它。

因为简单总比复杂好。

Answer 3

您可以使用 itertools.dropwhile 跳过所提供的函数返回 False 的项目，然后获取其余项目中的第一项（如果有）。如果您需要索引而不是项目，请合并 itertools 文档。

要反转提供的函数返回的真值，请使用 lambda (lambda x: not pred (x)，其中 pred 是提供的函数) 或命名包装器：

def negate(f):
    def wrapped(x):
        return not f(x)
    return wrapped

示例：

def odd(x): return x % 2 == 1
itertools.dropwhile(negate(odd), [2,4,1]).next()
# => 1

如果未找到匹配项，则会抛出 StopIteration ；将其包装在您自己的函数中以抛出您选择的异常。