Suchen Sie das erste Element in einer Sequenz, die einem Prädikat entspricht

Ich möchte auf idiomatische Weise das erste Element in einer Liste finden, das einem Prädikat entspricht.

Der aktuelle Code ist ziemlich hässlich:

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

Ich habe darüber nachgedacht, es zu ändern in:

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

Aber es muss etwas eleganteres geben ... Und es wäre schön, wenn es einen NoneWert zurückgibt , anstatt eine Ausnahme auszulösen, wenn keine Übereinstimmung gefunden wird.

Ich weiß, ich könnte einfach eine Funktion definieren wie:

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

Es ist jedoch geschmacklos, den Code mit solchen Dienstprogrammfunktionen zu füllen (und die Leute werden wahrscheinlich nicht bemerken, dass sie bereits vorhanden sind, sodass sie sich im Laufe der Zeit wiederholen), wenn es integrierte Funktionen gibt, die diese bereits bereitstellen.

So finden Sie das erste Element in einer Sequenz seq, die mit a übereinstimmt predicate:

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

Oder _{( itertools.ifilterauf Python 2)} :

next(filter(predicate, seq))

Es wird erhöht, StopIterationwenn es keine gibt.

Um zurückzukehren, Nonewenn es kein solches Element gibt:

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

Oder:

next(filter(predicate, seq), None)

Sie können einen Generatorausdruck mit einem Standardwert verwenden und dann next:

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

Obwohl Sie für diesen Einzeiler Python> = 2.6 verwenden müssen.

In diesem recht beliebten Artikel wird dieses Problem weiter erläutert: Sauberste Python-Funktion zum Suchen in der Liste? .

Ich glaube nicht, dass an beiden Lösungen, die Sie in Ihrer Frage vorgeschlagen haben, etwas falsch ist.

In meinem eigenen Code würde ich es jedoch so implementieren:

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

Die Syntax mit ()erstellt einen Generator, der effizienter ist als das gleichzeitige Generieren der gesamten Liste mit [].

Die Antwort von JF Sebastian ist am elegantesten, erfordert jedoch Python 2.6, wie fortran hervorhob.

Für Python-Version <2.6 ist hier das Beste, was ich mir vorstellen kann:

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

Wenn Sie später eine Liste benötigen (Liste behandelt die StopIteration) oder mehr als nur die erste, aber immer noch nicht alle benötigen, können Sie dies mit islice tun:

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

UPDATE: Obwohl ich persönlich eine vordefinierte Funktion namens first () verwende, die eine StopIteration abfängt und None zurückgibt, ist hier eine mögliche Verbesserung gegenüber dem obigen Beispiel: Vermeiden Sie die Verwendung von filter / ifilter:

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