Sobes.tech
Junior — Middle
64

Какие функции необходимо реализовать для создания собственного объекта итератора?

Answer from AI

sobes.tech AI

Для создания собственного объекта итератора в Python необходимо реализовать два метода:

  1. __iter__(self) — должен возвращать сам объект итератора. Обычно это return self.
  2. __next__(self) — возвращает следующий элемент последовательности. При достижении конца последовательности должен вызвать исключение StopIteration.

Пример:

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= len(self.data):
            raise StopIteration
        value = self.data[self.index]
        self.index += 1
        return value

# Использование
it = MyIterator([1, 2, 3])
for item in it:
    print(item)