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 docs 的“食谱”部分中的 enumerate。

要反转提供的函数返回的真值，请使用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；将其包装在您自己的函数中，以抛出您选择的异常。