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

Question

Domanda stupida in anticipo: voglio un modo idiomatico per trovare il primo elemento in un elenco che corrisponde a un predicato.

Il codice attuale è abbastanza brutto:

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

ho pensato di cambiare a:

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

ma ci deve essere qualcosa di più elegante ... e sarebbe bello se restituisce un valore None anziché generare un'eccezione se non viene trovata alcuna corrispondenza.

So che potrei solo definire una funzione come:

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

ma è del tutto insapore per iniziare a riempire il codice con funzioni di utilità come questo (e la gente probabilmente non si accorgono che sono già lì, in modo da tendono a essere ripetuti nel tempo) se ci sono insiemi integrati che forniscono già lo stesso.

Answer 1

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

solleva StopIteration se non c'è nessuno.

next(ifilter(predicate, seq), None)

rendimenti None se non c'è tale elemento.

Answer 2

È possibile utilizzare un generatore di espressione con un valore di default e poi next esso:

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

Anche se per questo one-liner è necessario essere utilizzando Python> = 2.6.

Questo articolo piuttosto popolare discute ulteriormente questo problema: Cleanest Python find-in-list function?.

Answer 3

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

Nel mio codice, vorrei implementare in questo modo però:

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

La sintassi con () crea un generatore, che è più efficiente di generare tutta la lista in una sola volta con [].

Answer 4

J.F. La risposta di Sebastian è molto elegante ma richiede python 2.6 come indicato da fortran.

Per versione Python < 2.6, ecco il meglio che posso venire con:

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

In alternativa, se avete bisogno di una lista più tardi (lista gestisce la StopIteration), o avete bisogno di più di solo il primo, ma ancora non è tutto, si può farlo con iSlice:

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

UPDATE: Anche se io sono personalmente usando una funzione predefinita chiamata prima() che cattura uno StopIteration e restituisce None, Ecco un possibile miglioramento rispetto l'esempio precedente: evitare di utilizzare il filtro/IFilter:

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