Perchè itertools.cycle di Python deve creare una copia del iterabile?

Question

La documentazione per itertools.cycle Python() fornisce un'implementazione pseudo-codice come:

def cycle(iterable): 
    # cycle('ABCD') --> A B C D A B C D A B C D ... 
    saved = [] 
    for element in iterable: 
     yield element 
     saved.append(element) 
    while saved: 
     for element in saved: 
       yield element 

seguito, si afferma: "Nota, questo membro del toolkit può richiedere notevole spazio di memoria (a seconda della lunghezza del l'iterabile). "

io in fondo stavo andando su questa strada, tranne che ho fatto questo, che non richiede la creazione di una copia del iterabile:

def loop(iterable): 
    it = iterable.__iter__() 

    while True: 
     try: 
      yield it.next() 
     except StopIteration: 
      it = iterable.__iter__() 
      yield it.next() 

x = {1, 2, 3} 

hard_limit = 6 
for i in loop(x): 
    if hard_limit <= 0: 
     break 

    print i 
    hard_limit -= 1 

stampe:

1 
2 
3 
1 
2 
3 

Sì, mi rendo conto che la mia implementazione non funzionerebbe per gli str, ma potrebbe essere fatto a. Sono più curioso del motivo per cui crea un'altra copia. Ho la sensazione che abbia a che fare con la garbage collection, ma non sono ben studiato in quest'area di Python.

Grazie!

Answer 1

Iterables può essere iterato solo su una volta.

Si crea un nuovo iterabile nel proprio ciclo. Il ciclo non può farlo, deve funzionare con qualsiasi cosa tu abbia passato. cycle non può semplicemente ricreare l'iterabile. È quindi costretto a memorizzare tutti gli elementi che produce l'iteratore originale.

Se si dovesse passare in seguito generatore, invece, il vostro loop() fallisce:

def finite_generator(source=[3, 2, 1]): 
    while source: 
     yield source.pop() 

Ora il vostro loop() produce:

>>> hard_limit = 6 
>>> for i in loop(finite_generator()): 
...  if hard_limit <= 0: 
...   break 
...  print i 
...  hard_limit -= 1 
... 
1 
2 
3 

Il tuo codice dovrebbe funzionare solo per le sequenze, per i quali utilizzando cycle() sarebbe eccessivo; in tal caso non è necessario il carico di archiviazione di cycle(). Semplificarlo a:

def loop_sequence(seq): 
    while True: 
     for elem in seq: 
      yield elem