Làm thế nào tôi có thể nhận được một trình tạo Python để trả về None chứ không phải StopIteration?

Question

Tôi đang sử dụng máy phát điện để thực hiện tìm kiếm trong danh sách như ví dụ này đơn giản:

>>> a = [1,2,3,4] 
>>> (i for i, v in enumerate(a) if v == 4).next() 
3 

(Chỉ cần khung ví dụ một chút, tôi đang sử dụng danh sách còn rất nhiều so với một ở trên, và các mục là một phức tạp hơn một chút so với int. Tôi làm theo cách này để toàn bộ danh sách sẽ không được duyệt qua mỗi khi tôi tìm kiếm chúng)

Bây giờ nếu thay vào đó, hãy đổi thành i == 666, nó sẽ trả lại StopIteration vì nó có thể ' t tìm thấy bất kỳ mục nhập 666 nào trong a.

Làm cách nào để tôi trả lại None? Tôi có thể tất nhiên quấn nó trong một điều khoản try ... except, nhưng là có một cách nhiều hơn pythonic để làm điều đó?

Answer 1

Nếu bạn đang sử dụng Python 2.6+ bạn nên sử dụng next built-in chức năng, không phải là phương pháp next (được thay thế bằng __next__ trong 3.x). Các next built-in có một đối số mặc định tùy chọn để quay trở lại nếu iterator là kiệt sức, thay vì nâng StopIteration:

next((i for i, v in enumerate(a) if i == 666), None) 

Answer 2

Bạn có thể chuỗi các máy phát điện với (Không,):

from itertools import chain 
a = [1,2,3,4] 
print chain((i for i, v in enumerate(a) if v == 6), (None,)).next() 

nhưng tôi nghĩ a.index (2) sẽ không đi qua danh sách đầy đủ, khi 2 được tìm thấy, việc tìm kiếm kết thúc. bạn có thể kiểm tra điều này:

>>> timeit.timeit("a.index(0)", "a=range(10)") 
0.19335955439601094 
>>> timeit.timeit("a.index(99)", "a=range(100)") 
2.1938486138533335