Rechercher le premier élément d'une séquence qui correspond à un prédicat

Je veux une manière idiomatique de trouver le premier élément d'une liste qui correspond à un prédicat.

Le code actuel est assez moche:

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

J'ai pensé à le changer en:

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

Mais il doit y avoir quelque chose de plus élégant ... Et ce serait bien s'il renvoie une Nonevaleur plutôt que de lever une exception si aucune correspondance n'est trouvée.

Je sais que je pourrais simplement définir une fonction comme:

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

Mais il est assez insipide de commencer à remplir le code avec des fonctions utilitaires comme celle-ci (et les gens ne remarqueront probablement pas qu'elles sont déjà là, donc elles ont tendance à se répéter au fil du temps) s'il y a des inserts intégrés qui fournissent déjà la même chose.

Pour rechercher le premier élément d'une séquence seqqui correspond à un predicate:

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

Ou _{( itertools.ifiltersur Python 2)} :

next(filter(predicate, seq))

Il augmente StopIterations'il n'y en a pas.

Pour retourner Nones'il n'y a pas un tel élément:

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

Ou:

next(filter(predicate, seq), None)

Vous pouvez utiliser une expression de générateur avec une valeur par défaut, puis next:

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

Bien que pour ce one-liner, vous devez utiliser Python> = 2.6.

Cet article plutôt populaire traite plus en détail de ce problème: Fonction de recherche dans la liste Python la plus propre? .

Je ne pense pas qu'il y ait quelque chose qui cloche dans les solutions que vous avez proposées dans votre question.

Dans mon propre code, je l'implémenterais comme ceci:

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

La syntaxe avec ()crée un générateur, ce qui est plus efficace que de générer toute la liste en même temps avec [].

La réponse de JF Sebastian est la plus élégante mais nécessite python 2.6 comme l'a souligné fortran.

Pour la version Python <2.6, voici le meilleur que je puisse proposer:

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

Alternativement, si vous aviez besoin d'une liste plus tard (la liste gère la StopIteration), ou si vous aviez besoin de plus que la première mais pas de la totalité, vous pouvez le faire avec islice:

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

MISE À JOUR: Bien que j'utilise personnellement une fonction prédéfinie appelée first () qui intercepte un StopIteration et renvoie None, voici une amélioration possible par rapport à l'exemple ci-dessus: évitez d'utiliser filter / ifilter:

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