Итераторы и итерируемые объекты

Итераторы и итерируемые объекты — это основные понятия Python, которые позволяют работать с последовательностями данных. Они лежат в основе циклов, включений и генераторов.

В этом уроке вы узнаете, как работают итераторы и итерируемые объекты и как создавать собственные итераторы.

Итерируемый объект — это любой объект в Python, элементы которого можно перебирать с помощью цикла for. К таким объектам относятся списки, кортежи, множества и словари.

# Example of an iterable
my_list = [1, 2, 3]
for item in my_list:
    print(item)  # Outputs: 1, 2, 3

Итератор — это объект, представляющий поток данных. Он реализует методы __iter__() и __next__().

# Example of an iterator
my_list = [1, 2, 3]
iterator = iter(my_list)
print(next(iterator))  # Outputs: 1
print(next(iterator))  # Outputs: 2
print(next(iterator))  # Outputs: 3

Итерируемые объекты можно перебирать, например списки, а итераторы выдают по одному элементу за раз с помощью метода next(). Итерируемый объект становится итератором, если передать его функции iter().

Вы можете создать пользовательские итераторы, определив класс с методами __iter__() и __next__():

# Custom iterator example
class Counter:
    def __init__(self, start, end):
        self.current = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.current > self.end:
            raise StopIteration
        self.current += 1
        return self.current - 1

counter = Counter(1, 5)
for number in counter:
    print(number)  # Outputs: 1, 2, 3, 4, 5

Функция iter() возвращает объект-итератор из итерируемого объекта:

# Using iter() on a list
my_list = [1, 2, 3]
iterator = iter(my_list)
print(next(iterator))  # Outputs: 1

Когда итератор исчерпан, он вызывает исключение StopIteration. Так Python сообщает о завершении последовательности.

# Handling StopIteration
my_list = [1, 2, 3]
iterator = iter(my_list)

while True:
    try:
        print(next(iterator))
    except StopIteration:
        break

Генераторы позволяют проще создавать итераторы. Они автоматически реализуют методы __iter__() и __next__().

# Example of a generator as an iterator
def my_generator():
    yield 1
    yield 2
    yield 3

for value in my_generator():
    print(value)  # Outputs: 1, 2, 3

Вот некоторые ошибки, которых следует избегать:

• Забывание реализовать __iter__() и __next__() в пользовательских итераторах.
• Некорректная обработка исключений StopIteration.
• Путаница между итераторами и итерируемыми объектами.

В этом уроке вы узнали:

• О различии между итерируемыми объектами и итераторами.
• Как использовать __iter__() и __next__() для создания пользовательских итераторов.
• Как использовать функцию iter() и обрабатывать исключения StopIteration.
• Как генераторы упрощают создание итераторов.

Отлично! Перейдём к следующей теме.