encuentra el primer elemento en una secuencia que coincide con un predicado

Question

Estúpida pregunta por delante: quiero una forma idiomática de encontrar el primer elemento en una lista que coincida con un predicado.

El código actual es bastante feo:

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

He pensado en cambiar a:

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

Pero tiene que haber algo más elegante ... y sería bueno si devuelve un valor de None en lugar de generar una excepción si no se encuentra ninguna coincidencia.

sé que sólo pudiera definir una función como:

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

pero es bastante insípido para comenzar a llenar el código con funciones de utilidad como esta (y la gente probablemente no se dan cuenta de que son ya allí, por lo tienden a repetirse a lo largo del tiempo) si hay ins integrados que ya proporcionan lo mismo.

Answer 1

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

Plantea StopIteration si no hay ninguno.

next(ifilter(predicate, seq), None)

vuelve None si no existe tal elemento.

Answer 2

Se puede usar un generador de expresión con un valor predeterminado y luego se next:

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

Aunque para esto de una sola línea que necesita utilizar Python> = 2.6.

Este artículo bastante popular discute aún más este problema: Cleanest Python find-in-list function?.

Answer 3

No creo que haya nada de malo con las soluciones que propuso en su pregunta.

En mi propio código, me gustaría ponerlo en práctica como esto sin embargo:

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

La sintaxis con () crea un generador, que es más eficiente que la generación de toda la lista a la vez con [].

Answer 4

J.F. La respuesta de Sebastian es muy elegante, pero requiere el pitón 2.6 como señaló fortran.

Para Python versión < 2.6, esto es lo mejor que puedo llegar a:

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

Alternativamente, si usted necesita una lista más adelante (la lista se encarga de la StopIteration), o si necesita algo más que la primera, pero no todos, puede hacerlo con iSlice:

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

ACTUALIZACIÓN: Aunque personalmente estoy usando una función predefinida llamada primero() que detecta un StopIteration y devuelve None, he aquí una posible mejora con respecto al ejemplo anterior: evitar el uso de filtros/ifilter:

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