Python | Вопросы собесов

Какие есть методы чтобы реализовать протокол итерирования данных ?

Спросят с вероятностью 3%



Для реализации протокола итерирования данных в Python необходимо использовать два метода: __iter__() и __next__().



Протокол итератора



1️⃣Метод `__iter__()`:

Этот метод должен возвращать объект-итератор. В простом случае он возвращает сам объект, если объект реализует метод __next__(). Метод __iter__() необходим для того, чтобы объект можно было использовать в конструкциях, которые требуют итерируемого объекта, таких как циклы for.



2️⃣Метод __next__():

Этот метод возвращает следующий элемент в последовательности. Когда элементы заканчиваются, метод должен вызвать исключение StopIteration для остановки итерации.



Рассмотрим пример класса, который реализует протокол итерирования:

class MyRange:

    def __init__(self, start, end):

        self.start = start

        self.end = end

        self.current = start



    def __iter__(self):

        self.current = self.start  # Перезапуск итератора при каждом вызове

        return self



    def __next__(self):

        if self.current >= self.end:

            raise StopIteration

        else:

            self.current += 1

            return self.current - 1



# Использование

for number in MyRange(1, 5):

    print(number)

В этом примере класс MyRange определяет простую последовательность чисел от start до end. Метод __iter__() возвращает сам объект, устанавливая начальное значение для итерации. Метод __next__() возвращает следующий элемент последовательности или вызывает исключение StopIteration, когда все элементы были возвращены.



Дополнительно: итераторы и генераторы



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



Пример:

def my_range(start, end):

    current = start

    while current < end:

        yield current

        current += 1



# Использование

for number in my_range(1, 5):

    print(number)

Генераторная функция my_range возвращает итератор, который выполняет ту же функцию, что и класс MyRange, но с более лаконичным синтаксисом.



Для реализации протокола итерирования в Python нужны методы __iter__() (возвращает итератор) и __next__() (возвращает следующий элемент или StopIteration). Это можно сделать через классы или генераторы.



👉 Можно посмотреть Примеры как отвечают люди на этот вопрос, или перейти К списку 1096 вопроса на Python разработчика. Ставь 👍 если нравится контент



🔐 База собесов | 🔐 База тестовых