Encuentre el primer elemento en una secuencia que coincida con un predicado

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 cambiarlo a:

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

Pero debe haber algo más elegante ... Y sería bueno si devuelve un Nonevalor en lugar de generar una excepción si no se encuentra una coincidencia.

Sé que podría 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 comenzar a llenar el código con funciones de utilidad como esta (y las personas probablemente no notarán que ya están allí, por lo que tienden a repetirse con el tiempo) si hay elementos integrados que ya proporcionan lo mismo.

Para buscar el primer elemento en una secuencia seqque coincida con predicate:

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

O _{( itertools.ifilteren Python 2)} :

next(filter(predicate, seq))

Se eleva StopIterationsi no hay ninguno.

Para volver Nonesi no hay tal elemento:

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

O:

next(filter(predicate, seq), None)

Puede usar una expresión generadora con un valor predeterminado y luego next:

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

Aunque para esta línea necesita usar Python> = 2.6.

Este artículo bastante popular discute más este problema: ¿La función de búsqueda en la lista de Python más limpia? .

No creo que haya nada malo con ninguna de las soluciones que propusiste en tu pregunta.

Sin embargo, en mi propio código, lo implementaría así:

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

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

La respuesta de JF Sebastian es muy elegante, pero requiere python 2.6, como señaló Fortran.

Para la versión Python <2.6, aquí está lo mejor que se me ocurre:

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

Alternativamente, si necesita una lista más tarde (la lista maneja StopIteration), o si necesita más que solo el primero pero aún no todo, 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 first () que detecta una StopIteration y devuelve None, aquí hay una posible mejora con respecto al ejemplo anterior: evite usar filter / ifilter:

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