W jaki sposób można uzyskać generator Python, aby zwrócić None, a nie StopIteration?

Question

używam generatorów do wyszukiwania w listach jak ten prosty przykład:

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

(Wystarczy wykadrować przykładzie trochę używam bardzo dużo dłuższych list w stosunku do powyższego, a wpisy są trochę bardziej skomplikowane niż int. robię to w ten sposób, więc cała listy nie będzie przesuwany za każdym razem, kiedy wyszukiwania nich)

teraz gdybym zamiast zmienić się i == 666, byłoby zwrócić StopIteration ponieważ może” t znaleźć dowolny wpis 666 w a.

W jaki sposób mogę go zwrócić zamiast numeru None? Mógłbym oczywiście zawinąć to w klauzulę try ... except, ale czy jest tam bardziej pythonic sposób to zrobić?

Answer 1

Jeśli używasz Python 2.6+ należy użyć next wbudowaną funkcję, a nie metoda next (który został zastąpiony __next__ w 3.x). next wbudowaną bierze domyślną opcjonalnego argumentu, aby powrócić jeśli iterator jest wyczerpany, zamiast podnoszenia StopIteration:

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

Answer 2

Można łańcucha z generatora (None,):

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

ale myślę a.index (2) nie będzie przemierzać całą listę, po 2 zostanie znaleziony, wyszukiwanie zostanie zakończone. można to sprawdzić:

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