अजगर में एक iterable (नहीं एक सूची) के पहले n आइटम

Question

मैं एक iterable (upd के पहले n से अधिक आइटम पुनरावृत्ति की एक pythonic रास्ता तलाश कर रहा हूँ से अधिक तेजी से बार-बार दोहराना सूचियों के लिए चीजें छोटी हैं), और जितनी जल्दी हो सके इसे करने के लिए यह काफी महत्वपूर्ण है। इस तरह मैं इसे अब करता हूं:

count = 0 
for item in iterable: 
do_something(item) 
count += 1 
if count >= n: break 

मुझे साफ नहीं लगता है। ऐसा करने का एक और तरीका है:

for item in itertools.islice(iterable, n): 
    do_something(item) 

यह अच्छा लग रहा है, सवाल यह काफी तेजी से कुछ जनरेटर (रों) के साथ उपयोग करने के लिए है? उदाहरण के लिए:

pair_generator = lambda iterable: itertools.izip(*[iter(iterable)]*2) 
for item in itertools.islice(pair_generator(iterable), n): 
so_something(item) 

क्या यह पहली विधि की तुलना में तेज़ी से दौड़ जाएगा? क्या ऐसा करने का कोई आसान तरीका है?

Answer 1

for item in itertools.islice(iterable, n): सबसे स्पष्ट, आसान तरीका है इसे करने के लिए। यह मनमाने ढंग से पुनरावृत्तियों के लिए काम करता है और ओ (एन) है, जैसे किसी भी सोल समाधान होगा।

यह कल्पना की जा सकती है कि एक और समाधान बेहतर प्रदर्शन कर सकता है; हम समय के बिना नहीं पता होगा। मैं समय के साथ परेशान करने की अनुशंसा नहीं करता जबतक कि आप profile अपना कोड न दें और यह कॉल हॉटस्पॉट बनें। जब तक यह एक आंतरिक पाश के भीतर buries नहीं है, यह बहुत संदिग्ध है कि यह होगा। सभी बुराईयो की जड़ समयपूर्व इष्टतमीकरण है।

---

अगर मैं था वैकल्पिक समाधान के लिए लग रहा है, मैं for count, item in enumerate(iterable): if count > n: break ... और for i in xrange(n): item = next(iterator) ... की तरह लोगों पर विचार करेंगे। मुझे लगता है कि इससे मदद मिलेगी, लेकिन अगर हम वास्तव में चीजों की तुलना करना चाहते हैं तो वे कोशिश करने लायक हैं। अगर मैं ऐसी परिस्थिति में फंस गया था जहां मैंने प्रोफाइल किया और पाया कि यह एक आंतरिक लूप में एक हॉटस्पॉट था (क्या यह वास्तव में आपकी स्थिति है?), मैं islice वैश्विक iterools की विशेषता प्राप्त करने से नाम लुकअप को भी कम करने का प्रयास करूंगा फ़ंक्शन को स्थानीय नाम पर पहले से बाध्य करना।

ये चीजें हैं जो आप साबित करने के बाद ही करते हैं कि वे आपकी सहायता करेंगे। लोग उन्हें कई बार करने की कोशिश करते हैं। यह उनके कार्यक्रमों को तेजी से तेज़ी से बनाने में मदद नहीं करता है; यह सिर्फ अपने कार्यक्रमों को और खराब बनाता है।

Answer 2

सूची में से एक? प्रयास करें

for k in mylist[0:n]: 
    # do stuff with k 

आप भी एक समझ अगर आप

my_new_list = [blah(k) for k in mylist[0:n]] 

Answer 3

की जरूरत का उपयोग कर सकते हैं यह एक सूची है तो आप उपयोग कर सकते हैं टुकड़ा करने की क्रिया है:

list[:n] 

Answer 4

itertools सीधे लागू होने पर सबसे तेज़ समाधान होता है।

जाहिर है, की जांच करने के लिए एक ही रास्ता बेंचमार्क है - जैसे, aaa.py

import itertools 

def doit1(iterable, n, do_something=lambda x: None): 
    count = 0 
    for item in iterable: 
    do_something(item) 
    count += 1 
    if count >= n: break 

def doit2(iterable, n, do_something=lambda x: None): 
    for item in itertools.islice(iterable, n): 
     do_something(item) 

pair_generator = lambda iterable: itertools.izip(*[iter(iterable)]*2) 

def dd1(itrbl=range(44)): doit1(itrbl, 23) 
def dd2(itrbl=range(44)): doit2(itrbl, 23) 

में बचाने के लिए और देखें ...:

$ python -mtimeit -s'import aaa' 'aaa.dd1()' 
100000 loops, best of 3: 8.82 usec per loop 
$ python -mtimeit -s'import aaa' 'aaa.dd2()' 
100000 loops, best of 3: 6.33 usec per loop 

इतनी स्पष्ट रूप से, itertools तेजी से यहाँ है - सत्यापित करने के लिए अपने स्वयं के डेटा के साथ बेंचमार्क।

बीटीडब्ल्यू, मुझे timeit कमांड लाइन से अधिक उपयोग करने योग्य लगता है, इसलिए मैं हमेशा इसका उपयोग कैसे करता हूं - फिर यह उस गति के सही "गति का क्रम" चलाता है जिसकी आप विशेष रूप से मापने की कोशिश कर रहे हैं , उन 10, 100, 1000, और इतने पर - यहां, एक माइक्रोसेकंद और अंतर का आधा अंतर करने के लिए, एक सौ हजार लूप सही हैं।

Answer 5

आप enumerate उपयोग कर सकते हैं मूलतः एक ही पाश आपके पास लिखने के लिए, लेकिन एक और अधिक सरल, pythonic तरह से:

 
for idx, val in enumerate(iterableobj): 
    if idx > n: 
     break 
    do_something(val)