define simple iterators in python code example

Example 1: what is iterator in python

An iterator is an object that can be iterated upon, meaning that you can traverse through all the values.
#EXAMPLE OF ITERATOR
nums=[2,3,4,5,6,7]
it=iter(nums)
print(it.__next__())
for i in it:
  print(i)

Example 2: iterator in python

class PowTwo:
    """Class to implement an iterator
    of powers of two"""

    def __init__(self, max=0):
        self.max = max

    def __iter__(self):
        self.n = 0
        return self

    def __next__(self):
        if self.n <= self.max:
            result = 2 ** self.n
            self.n += 1
            return result
        else:
            raise StopIteration


# create an object
numbers = PowTwo(3)

# create an iterable from the object
i = iter(numbers)

# Using next to get to the next iterator element
print(next(i))
print(next(i))
print(next(i))
print(next(i))
print(next(i))

Tags:

Misc Example