जेनरेट किए गए अनुक्रम को कैसे पता चलेगा एक निश्चित लंबाई

Question

मैं जानना चाहता हूं कि जेनरेट किए गए अनुक्रम में 2 से कम प्रविष्टियां हैं या नहीं।

>>> secret = 5 
>>> len(list(sequence())) < 2 
True 

जाहिर है, इस पूरे जनरेटर की खपत:

>>> def sequence(): 
...  for i in xrange(secret): 
...   yield i 

मेरे अक्षम विधि सूची बनाते हैं, और इसकी लंबाई को मापने के लिए है।

मेरे असली मामले में जनरेटर एक बड़े नेटवर्क को पार कर सकता है। मैं पूरे जेनरेटर का उपभोग किए बिना चेक करना चाहता हूं, या बड़ी सूची बनाना चाहता हूं।

एक recipe in the itertools documentation नहीं है:

def take(n, iterable): 
    "Return first n items of the iterable as a list" 
    return list(islice(iterable, n)) 

यह केवल अधिकतम लंबाई n है, जो बेहतर है की एक सूची बनाता है।

तो मैं कह सकते हैं:

>>> len(take(2, sequence()) < 2 

वहाँ एक और भी अधिक pythonic, कुशल यह करने के लिए रास्ता नहीं है?

Answer 1

take का उपयोग कर समाधान islice का उपयोग करता है, हम sum उपयोग कर सकते हैं

>>> from itertools import islice 
>>> len(list(islice(sequence(), 2)) 
2 

सूची बनाने से बचने के लिए: एक सूची बनाता है और यह की लंबाई लेता

+०१२३५१६४१०६

>>> timeit('len(list(islice(xrange(1000), 2)))', 'from itertools import islice') 
1.089650974650752 

>>> timeit('sum(1 for _ in islice(xrange(1000), 2))', 'from itertools import islice') 
0.7579448552500647 

इसे रैपिंग:

>>> sum(1 for _ in islice(sequence(), 2) 
2 

इस बार का लगभग 70% लेता है

>>> def at_most(n, elements): 
...  return sum(1 for _ in islice(elements, n + 1)) <= n 

>>> at_most(5, xrange(5)) 
True 

>>> at_most(2, xrange(5)) 
False 

Answer 2

पायथन 3.4 के रूप में, जेनरेटर length hint लागू कर सकते हैं। यदि कोई जनरेटर इसे लागू करता है तो इसे object.__length_hint__() method के माध्यम से उजागर किया जाएगा।

आप इसके लिए operator.length_hint() function के साथ परीक्षण कर सकते हैं।

अगर ऐसा है उपलब्ध नहीं, अपने ही एकमात्र विकल्प तत्वों का उपभोग करने के लिए है, और take() नुस्खा के आपके उपयोग करने के लिए सबसे कारगर तरीका है:

from operator import length_hint 
from itertools import chain 

elements = [] 
length = length_hint(gen, None) 
if length is None: 
    elements = list(take(2, gen)) 
    length = len(elements) 
if length >= 2: 
    # raise an error 
# use elements, then gen 
gen = chain(elements, gen)