Temukan elemen pertama dalam urutan yang cocok dengan predikat

Saya ingin cara idiomatis untuk menemukan elemen pertama dalam daftar yang cocok dengan predikat.

Kode saat ini sangat jelek:

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

Saya sudah berpikir untuk mengubahnya menjadi:

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

Tapi pasti ada sesuatu yang lebih elegan ... Dan alangkah baiknya jika mengembalikan Nonenilai daripada menaikkan pengecualian jika tidak ada kecocokan yang ditemukan.

Saya tahu saya bisa mendefinisikan fungsi seperti:

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

Tetapi cukup hambar untuk mulai mengisi kode dengan fungsi-fungsi utilitas seperti ini (dan orang-orang mungkin tidak akan menyadari bahwa mereka sudah ada di sana, sehingga mereka cenderung untuk diulangi dari waktu ke waktu) jika ada seluk-beluk internal yang sudah menyediakan hal yang sama.

Untuk menemukan elemen pertama dalam urutan seqyang cocok dengan predicate:

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

Atau _{( itertools.ifilterdi Python 2)} :

next(filter(predicate, seq))

Itu memunculkan StopIterationjika tidak ada.

Untuk kembali Nonejika tidak ada elemen seperti itu:

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

Atau:

next(filter(predicate, seq), None)

Anda bisa menggunakan ekspresi generator dengan nilai default dan kemudian next:

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

Meskipun untuk one-liner ini Anda harus menggunakan Python> = 2.6.

Artikel yang agak populer ini lebih lanjut membahas masalah ini: Fungsi find-in-list Python terbersih? .

Saya tidak berpikir ada yang salah dengan solusi yang Anda ajukan dalam pertanyaan Anda.

Dalam kode saya sendiri, saya akan mengimplementasikannya seperti ini:

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

Sintaks dengan ()menciptakan generator, yang lebih efisien daripada menghasilkan semua daftar sekaligus [].

Jawaban JF Sebastian paling elegan tetapi membutuhkan python 2.6 seperti yang ditunjukkan fortran.

Untuk versi Python <2.6, ini yang terbaik yang bisa saya lakukan:

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

Atau jika Anda membutuhkan daftar nanti (daftar menangani StopIteration), atau Anda membutuhkan lebih dari sekadar yang pertama tetapi masih belum semuanya, Anda dapat melakukannya dengan islice:

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

UPDATE: Meskipun saya secara pribadi menggunakan fungsi yang telah ditentukan yang disebut first () yang menangkap StopIteration dan mengembalikan None, Berikut adalah kemungkinan peningkatan atas contoh di atas: hindari menggunakan filter / ifilter:

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