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

Question

Câu hỏi ngu ngốc phía trước: 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 vị từ.

Mã hiện nay là khá xấu xí:

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

Tôi đã suy nghĩ về việc thay đổi nó để:

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

Nhưng phải có một cái gì đó tao nhã hơn ... Và nó sẽ được tốt đẹp nếu nó trả về giá trị None thay vì tăng ngoại lệ nếu không tìm thấy kết quả phù hợp.

Tôi biết tôi chỉ có thể xác định 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 nó là khá nhạt để bắt đầu điền mã với chức năng tiện ích như thế này (và những người có thể sẽ không nhận thấy rằng họ đã có, vì vậy chúng có xu hướng được lặp đi lặp lại theo thời gian) nếu có các bản dựng sẵn được cung cấp giống nhau.

Answer 1

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

Nó làm tăng StopIteration nếu có ai sánh kịp.

next(ifilter(predicate, seq), None)

lợi nhuận None nếu không có yếu tố như vậy.

Answer 2

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

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

Mặc dù cho điều này một lót bạn cần phải 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: Cleanest Python find-in-list function?.

Answer 3

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

Trong mã của riêng tôi, tôi sẽ thực hiện nó như mặc dù điều này:

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

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

Answer 4

J.F. Câu trả lời của Sebastian là thanh lịch nhất nhưng yêu cầu python 2.6 như fortran chỉ ra.

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

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

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

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

UPDATE: Mặc dù tôi cá nhân sử dụng một chức năng được xác định trước gọi là đầu tiên() mà bắt một StopIteration và trả về Không, đây là một sự cải thiện tốt hơn 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()