Какие методы нужно определить, чтобы создать собственный итератор

Чтобы создать собственный итератор в Python, нужно определить два метода в классе: __iter__() и __next__().

• __iter__() должен возвращать сам объект итератора (обычно self).
• __next__() должен возвращать следующее значение или вызывать StopIteration, если элементы закончились.

Пример:

class CountDown:
    def __init__(self, start):
        self.current = start

    def __iter__(self):
        return self

    def __next__(self):
        if self.current <= 0:
            raise StopIteration
        num = self.current
        self.current -= 1
        return num

for num in CountDown(3):
    print(num)  # 3, 2, 1

Если нужен только итерируемый объект (без сохранения состояния), можно использовать yield в __iter__().