Znajdź pierwszy element w sekwencji, który pasuje do predykatu

Chcę idiomatycznego sposobu na znalezienie pierwszego elementu na liście, który pasuje do predykatu.

Obecny kod jest dość brzydki:

[x for x in seq if predicate(x)][0]

Myślałem o zmianie tego na:

from itertools import dropwhile
dropwhile(lambda x: not predicate(x), seq).next()

Ale musi być coś bardziej eleganckiego ... I byłoby miło, gdyby zwracało Nonewartość, zamiast zgłaszać wyjątek, jeśli nie zostanie znalezione dopasowanie.

Wiem, że mógłbym po prostu zdefiniować funkcję taką jak:

def get_first(predicate, seq):
    for i in seq:
        if predicate(i): return i
    return None

Ale rozpoczęcie wypełniania kodu takimi funkcjami użytkowymi jest zupełnie bez smaku (a ludzie prawdopodobnie nie zauważą, że już tam są, więc mają tendencję do powtarzania się w czasie), jeśli istnieją wbudowane narzędzia, które już zapewniają to samo.

Aby znaleźć pierwszy element w sekwencji, seqktóry pasuje do a predicate:

next(x for x in seq if predicate(x))

Lub _{( itertools.ifilterw Pythonie 2)} :

next(filter(predicate, seq))

Podnosi się, StopIterationjeśli nie ma.

Aby wrócić, Nonejeśli nie ma takiego elementu:

next((x for x in seq if predicate(x)), None)

Lub:

next(filter(predicate, seq), None)

Możesz użyć wyrażenia generatora z wartością domyślną, a następnie next:

next((x for x in seq if predicate(x)), None)

Chociaż w przypadku tego jednowierszowego kodu musisz używać Pythona> = 2.6.

Ten dość popularny artykuł dalej omawia ten problem: Najczystsza funkcja znajdowania na liście w Pythonie? .

Nie sądzę, żeby było coś złego w obu rozwiązaniach, które zaproponowałeś w swoim pytaniu.

We własnym kodzie zaimplementowałbym to jednak w następujący sposób:

(x for x in seq if predicate(x)).next()

Składnia with ()tworzy generator, który jest bardziej wydajny niż generowanie całej listy naraz [].

Odpowiedź JF Sebastiana jest najbardziej elegancka, ale wymaga Pythona 2.6, jak wskazał Fortran.

Dla wersji Pythona <2.6, oto najlepsze, co mogę wymyślić:

from itertools import repeat,ifilter,chain
chain(ifilter(predicate,seq),repeat(None)).next()

Alternatywnie, jeśli potrzebowałeś listy później (lista obsługuje StopIteration) lub potrzebujesz czegoś więcej niż tylko pierwszego, ale nadal nie wszystkiego, możesz to zrobić za pomocą islice:

from itertools import islice,ifilter
list(islice(ifilter(predicate,seq),1))

AKTUALIZACJA: Chociaż osobiście używam predefiniowanej funkcji o nazwie first (), która przechwytuje StopIteration i zwraca None, Oto możliwe ulepszenie w stosunku do powyższego przykładu: unikaj używania filtra / ifilter:

from itertools import islice,chain
chain((x for x in seq if predicate(x)),repeat(None)).next()