Python - функциональный "find"?

Question

Мне нужна функция, которая способна выполнять итерацию по коллекции, вызывая предоставленную функцию с элементом коллекции в качестве параметра и возвращая параметр или индекс при получении «True» из предоставленной функции.

Он somethong так:

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 

Так мне интересно, есть ли что-нибудь подобное в стандартном наборе инструментов питона?

Answer 1

Вы можете использовать itertools.dropwhile, чтобы пропустить пункты, для которых возвращается функция False, а затем взять первый элемент остального (если есть). Если вам нужен индекс, а не элемент, включите enumerate из раздела «Рецепты» itertools docs.

Чтобы изменить значения истинности, возвращенные в комплект поставки функции, использовать 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 

Это будет throw StopIteration, если соответствующий элемент не найден; оберните его в свою собственную функцию, чтобы вместо этого выбрать исключение из вашего выбора.

Answer 2

Пробег: itertools и, например, ifilter.

Answer 3

Я не думаю, что есть такая функция с такой точной семантикой, и в любом случае ваша функция коротка, достаточно хороша, и вы можете легко ее улучшить для последующего использования, поэтому используйте ее.

потому что простой лучше, чем сложный.