What are some examples of iterators in Python?

What are some examples of iterators in Python?

In Python, iterators are objects that allow you to traverse through a collection. Examples include lists, tuples, and dictionaries.

Can you show me how to create a custom iterator?

Sure! You can create a custom iterator by defining a class with __iter__() and __next__() methods. Here's a simple example:

What does the __iter__() method do?

The __iter__() method returns the iterator object itself. It's required for the object to be an iterator.

Can you give me a code example of a custom iterator?

Absolutely! Here's a simple custom iterator that counts from 1 to a specified number:

class Count:
    def __init__(self, max):
        self.max = max
        self.current = 1
    
    def __iter__(self):
        return self
    
    def __next__(self):
        if self.current > self.max:
            raise StopIteration
        else:
            current = self.current
            self.current += 1
            return current