Tìm phần tử đầu tiên trong chuỗi khớp với một vị ngữ

Tôi muốn một cách thành ngữ để tìm phần tử đầu tiên trong danh sách khớp với một vị ngữ.

Mã hiện tại khá xấu:

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

Tôi đã nghĩ về việc thay đổi nó thành:

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

Nhưng phải có một cái gì đó thanh lịch hơn ... Và sẽ thật tuyệt nếu nó trả về một Nonegiá trị thay vì đưa ra một ngoại lệ nếu không tìm thấy kết quả khớp nào.

Tôi biết tôi chỉ có thể định nghĩa một chức năng như:

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

Nhưng thật vô vị khi bắt đầu điền mã với các hàm tiện ích như thế này (và mọi người có thể sẽ không nhận thấy rằng chúng đã ở đó, vì vậy chúng có xu hướng được lặp lại theo thời gian) nếu có các phần mềm được xây dựng tương tự.

Để tìm phần tử đầu tiên trong chuỗi seqkhớp với predicate:

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

Hoặc _{( itertools.ifiltertrên Python 2)} :

next(filter(predicate, seq))

Nó tăng StopIterationnếu không có.

Để trở về Nonenếu không có yếu tố đó:

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

Hoặc là:

next(filter(predicate, seq), None)

Bạn có thể sử dụng biểu thức trình tạo với giá trị mặc định và sau nextđó:

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

Mặc dù đối với lớp lót này, bạn cần sử dụng Python> = 2.6.

Bài viết khá phổ biến này thảo luận thêm về vấn đề này: Chức năng tìm trong danh sách Python sạch nhất? .

Tôi không nghĩ có gì sai với bất kỳ giải pháp nào bạn đề xuất trong câu hỏi của bạn.

Trong mã của riêng tôi, tôi sẽ thực hiện nó như thế này:

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

Cú pháp với việc ()tạo một trình tạo, hiệu quả hơn so với việc tạo tất cả danh sách cùng một lúc [].

Câu trả lời của JF Sebastian là thanh lịch nhất nhưng đòi hỏi python 2.6 như fortran đã chỉ ra.

Đối với phiên bản Python <2.6, đây là phiên bản tốt nhất tôi có thể nghĩ ra:

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

Ngoài ra, nếu bạn cần một danh sách sau này (danh sách xử lý StopIteration) hoặc bạn cần nhiều hơn chỉ là đầu tiên nhưng vẫn không phải là tất cả, bạn có thể thực hiện với islice:

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

CẬP NHẬT: Mặc dù cá nhân tôi đang sử dụng một hàm được xác định trước có tên là First () để bắt StopIteration và trả về Không, đây là một cải tiến có thể có trong ví dụ trên: tránh sử dụng bộ lọc / ifilter:

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