Jak rozpakować Iterator?

Question

Biorąc pod uwagę listę par xys, idiom Python rozpakować go do dwóch list jest:

xs, ys = zip(*xys) 

Jeśli xys jest iterator, jak mogę rozpakować go na dwa iteratory, bez zapisywania wszystkiego w pamięci?

Answer 1

Załóżmy, że masz jakieś iterowalny par:

a = zip(range(10), range(10)) 

Jeśli mam poprawnie interpretując co prosicie, można generować niezależnych iteratorów dla nowatorskich rozwiązań i sekund używając itertools.tee:

xs, ys = itertools.tee(a) 
xs, ys = (x[0] for x in xs), (y[1] for y in ys) 

Notatka to zapisze w pamięci "różnicę" między tym, jak jeden z nich się sprawdza w stosunku do drugiego.

Answer 2

Jeśli chcesz spożywać jeden iterator niezależnie od drugiego, nie ma sposobu, aby uniknąć wciągnięcia rzeczy do pamięci, ponieważ jeden z iteratorów będzie się rozwijał, podczas gdy drugi nie będzie (a zatem musi buforować).

Coś takiego pozwala iterować zarówno „rzeczy pozostawione” i „prawych pozycji” par:

import itertools 
import operator 

it1, it2 = itertools.tee(xys) 
xs = map(operator.itemgetter(0), it1)) 
ys = map(operator.itemgetter(1), it2)) 

print(next(xs)) 
print(next(ys)) 

... ale należy pamiętać, że jeśli zużywają tylko jedną iterację, drugi buforuje elementy w pamięci, dopóki ich nie zużyjesz.

(Btw, zakładając Python 3. W Pythonie 2 trzeba użyć itertools.imap(), nie map().)