ค้นหาองค์ประกอบแรกในลำดับที่ตรงกับเพรดิเคต

ฉันต้องการวิธีใช้สำนวนเพื่อค้นหาองค์ประกอบแรกในรายการที่ตรงกับภาคแสดง

รหัสปัจจุบันค่อนข้างน่าเกลียด:

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

ฉันคิดว่าจะเปลี่ยนเป็น:

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

แต่จะต้องมีสิ่งที่สวยงามกว่านี้ ... และมันจะดีถ้ามันคืนNoneค่าแทนที่จะยกข้อยกเว้นหากไม่พบคู่ที่ตรงกัน

ฉันรู้ว่าฉันสามารถกำหนดฟังก์ชันเช่น:

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

แต่ค่อนข้างจืดชืดที่จะเริ่มต้นกรอกรหัสด้วยฟังก์ชั่นยูทิลิตี้เช่นนี้ (และคนอาจจะไม่สังเกตเห็นว่าพวกเขามีอยู่แล้วดังนั้นพวกเขาจึงมักจะทำซ้ำในช่วงเวลา) หากมีการสร้างอินที่ให้ไว้เหมือนกัน

ในการค้นหาองค์ประกอบแรกในลำดับseqที่ตรงกับpredicate:

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

หรือ_{( itertools.ifilterบน Python 2)} :

next(filter(predicate, seq))

มันจะเพิ่มขึ้นStopIterationหากไม่มี

หากต้องการส่งคืนNoneหากไม่มีองค์ประกอบดังกล่าว:

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

หรือ:

next(filter(predicate, seq), None)

คุณสามารถใช้นิพจน์ตัวสร้างกับค่าเริ่มต้นแล้วnext:

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

แม้ว่าสำหรับสายการบินเดียวคุณต้องใช้ Python> = 2.6

บทความยอดนิยมนี้ค่อนข้างพูดถึงปัญหานี้เพิ่มเติม: ฟังก์ชั่น Clean-Python ค้นหาในรายการ? .

ฉันไม่คิดว่าจะมีอะไรผิดปกติกับวิธีแก้ไขปัญหาที่คุณเสนอในคำถามของคุณ

ในรหัสของฉันฉันจะใช้มันเช่นนี้แม้ว่า:

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

ไวยากรณ์กับ()สร้างตัวสร้างซึ่งมีประสิทธิภาพมากกว่าการสร้างรายการทั้งหมดพร้อม[]กัน

คำตอบของเจเอฟเจบาสเตียนนั้นสง่างามที่สุด แต่ต้องการงูใหญ่ 2.6 ตามที่ Fortran ชี้ให้เห็น

สำหรับ Python เวอร์ชัน <2.6 นี่คือสิ่งที่ดีที่สุดที่ฉันจะได้รับ:

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

อีกทางเลือกหนึ่งถ้าคุณต้องการรายการในภายหลัง (รายการจัดการ StopIteration) หรือคุณต้องการมากกว่าแค่รายการแรก แต่ก็ยังไม่ครบทั้งหมดคุณสามารถทำได้ด้วย islice:

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

UPDATE: แม้ว่าฉันจะใช้ฟังก์ชั่นที่กำหนดไว้ล่วงหน้าที่เรียกว่า first () เป็นการส่วนตัวที่จับ StopIteration และคืนค่า None นี่คือการปรับปรุงที่เป็นไปได้สำหรับตัวอย่างข้างต้น: หลีกเลี่ยงการใช้ filter / ifilter:

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