स्टॉपइटरेशन के बजाए कोई भी वापस करने के लिए पाइथन जनरेटर कैसे प्राप्त कर सकता हूं?

Question

मैं जनरेटर का उपयोग कर रहा इस सरल उदाहरण की तरह खोजें सूचियों में प्रदर्शन करने के लिए:

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

(बस उदाहरण थोड़ा फ्रेम करने के लिए, मैं ऊपर एक की तुलना में बहुत ज्यादा देर सूचियों का उपयोग कर रहा हूँ, और प्रविष्टियों एक हैं थोड़ा और अधिक जटिल int से। मैं इसे इस तरह से तो पूरे सूचियों अब हर बार मैं उन्हें खोज)

चल नहीं की जाएगी करना अगर मैं बजाय i == 666 है कि बदल जाएगा, यह एक StopIteration वापसी होगी क्योंकि यह कर सकते हैं ' में कोई भी 666 प्रविष्टि नहीं ढूंढें।

मैं इसे कैसे None बजाय वापसी कर सकते हैं? मैं इसे try ... except खंड में लपेट सकता हूं, लेकिन क्या ऐसा करने के लिए एक और अधिक पागल तरीका है?

Answer 1

आप अजगर उपयोग कर रहे हैं 2.6+ आप next में निर्मित समारोह, नहीं next विधि (जो 3.x में __next__ से बदला गया) का उपयोग करना चाहिए। next में निर्मित एक वैकल्पिक डिफ़ॉल्ट तर्क लेता है, तो इटरेटर समाप्त हो रहा है वापस जाने के लिए, बजाय StopIteration जुटाने की:

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

Answer 2

आप (कोई नहीं,) के साथ जनरेटर श्रृंखला कर सकते हैं:

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

लेकिन मुझे लगता है a.index (2) पूरी सूची है, जब 2 पाया जाता है, खोज समाप्त हो गया है पार नहीं होंगे। आप इस परीक्षण कर सकते हैं:

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