Введение в итераторы и итерируемые объекты

Итераторы и итерируемые объекты — фундаментальные понятия 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()

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

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

Обработка StopIteration

Когда итератор исчерпан, он возбуждает исключение 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.
• Как генераторы упрощают создание итераторов.

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