Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to create a copy of python iterator? [duplicate]

In python i tried to create a copy of my iterator with using assignment however it create a copy of iterator which reference to the original iterator itself. For instance:

my_list = [5, 4, 3,2] first_it = iter(my_list) second_it = first_it  print next(first_it )        #it will print 5 print next(second_it)        #it will print 4 print next(first_it )        #it will print 3 

As you see in the example first_it and second_it both refer to same iterator object. Is it possible to create a copy of iterator object which is not reference to the original object?

Note This question is about how to creating a copy of iterator object by value. So don't mention for item in my_list: like solutions.
Thanks in advance

like image 598
Harun ERGUL Avatar asked Feb 09 '17 09:02

Harun ERGUL


People also ask

Can I copy iterator?

There's no way one could copy the current state of an iterator. The same happens to our original generator + promise usage. We could use the same syntax to work with events (i.e. repetitive callback calls), not just series of nested callbacks, but we don't have that vital iter. copy() method.

How do I use an iterator in python?

Create an IteratorTo create an object/class as an iterator you have to implement the methods __iter__() and __next__() to your object. As you have learned in the Python Classes/Objects chapter, all classes have a function called __init__() , which allows you to do some initializing when the object is being created.


1 Answers

Use the itertools.tee() function to produce copies; these use a buffer to share results between different iterators:

from itertools import tee  my_list = [5, 4, 3,2] first_it = iter(my_list) first_it, second_it = tee(first_it) print next(first_it)   # prints 5 print next(second_it)  # prints 5 print next(first_it)   # prints 4 

Note that you should no longer use the original iterator; use only the tees.

Note that the buffer also means that these can incur a significant memory cost if you advance one of the copies far ahead of the others! From the documentation:

This itertool may require significant auxiliary storage (depending on how much temporary data needs to be stored). In general, if one iterator uses most or all of the data before another iterator starts, it is faster to use list() instead of tee().

like image 135
Martijn Pieters Avatar answered Oct 08 '22 22:10

Martijn Pieters