Trova il primo elemento in una sequenza che corrisponde a un predicato

Voglio un modo idiomatico per trovare il primo elemento in un elenco che corrisponda a un predicato.

Il codice attuale è abbastanza brutto:

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

Ho pensato di cambiarlo in:

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

Ma ci deve essere qualcosa di più elegante ... E sarebbe bello se restituisse un Nonevalore piuttosto che sollevare un'eccezione se non viene trovata alcuna corrispondenza.

So che potrei semplicemente definire una funzione come:

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

Ma è abbastanza insipido iniziare a riempire il codice con funzioni di utilità come questa (e probabilmente le persone non noteranno che sono già lì, quindi tendono a ripetersi nel tempo) se ci sono insiemi che forniscono già lo stesso.

Per trovare il primo elemento in una sequenza seqche corrisponde a predicate:

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

Oppure _{( itertools.ifiltersu Python 2)} :

next(filter(predicate, seq))

Si alza StopIterationse non ce n'è.

Per restituire Nonese non esiste tale elemento:

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

O:

next(filter(predicate, seq), None)

È possibile utilizzare un'espressione del generatore con un valore predefinito e quindi next:

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

Sebbene per questo one-liner devi usare Python> = 2.6.

Questo articolo piuttosto popolare discute ulteriormente questo problema: la funzione di ricerca nell'elenco Python più pulita? .

Non credo che ci sia qualcosa di sbagliato in entrambe le soluzioni che hai proposto nella tua domanda.

Nel mio codice, lo implementerei in questo modo però:

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

La sintassi con ()crea un generatore, che è più efficiente che generare tutto l'elenco contemporaneamente [].

La risposta di JF Sebastian è molto elegante ma richiede python 2.6 come ha sottolineato Fortran.

Per la versione Python <2.6, ecco il meglio che posso inventare:

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

In alternativa, se hai bisogno di un elenco in seguito (l'elenco gestisce StopIteration) o hai bisogno di qualcosa di più del primo ma ancora non tutto, puoi farlo con islice:

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

AGGIORNAMENTO: Anche se sto personalmente usando una funzione predefinita chiamata first () che rileva StopIteration e restituisce None, ecco un possibile miglioramento rispetto all'esempio sopra: evita di usare filter / ifilter:

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