Soluzione pitonica per eliminare N valori da un iteratore

Question

Esiste una soluzione pitonica per eliminare i valori n da un iteratore? È possibile farlo da solo scartando n valori come segue:

def _drop(it, n): 
    for _ in xrange(n): 
     it.next() 

ma questo non è IMO elegante come codice Python dovrebbe essere. C'è un approccio migliore che mi manca qui?

Answer 1

Credo che si sta cercando la ricetta "consumare"

http://docs.python.org/library/itertools.html#recipes

def consume(iterator, n): 
    "Advance the iterator n-steps ahead. If n is none, consume entirely." 
    # Use functions that consume iterators at C speed. 
    if n is None: 
     # feed the entire iterator into a zero-length deque 
     collections.deque(iterator, maxlen=0) 
    else: 
     # advance to the empty slice starting at position n 
     next(islice(iterator, n, n), None) 

Se non è necessario il comportamento speciale quando n è None, si può semplicemente utilizzare

next(islice(iterator, n, n), None) 

Answer 2

È possibile creare una sezione iterativa che inizia dall'elemento n:

import itertools 
def drop(it, n): 
    return itertools.islice(it, n, None) 

Answer 3

Si potrebbe fare questo con l'uso di fantasia di itertools.dropwhile, ma io esiterei a chiamarlo in alcun modo elegante:

def makepred(n): 
    def pred(x): 
     pred.count += 1 
     return pred.count < n 
    pred.count = 0 
    return pred 

itertools.dropwhile(it, makepred(5)) 

davvero non consiglio questo, anche se - basandosi su gli effetti collaterali di una funzione di predicato sono molto sul lato dispari.