Sobes.tech
Middle

O que deve estar presente num objeto para que seja um iterador em Python?

sobes.tech IA

Resposta da IA

Um objeto é um iterador em Python se implementar dois métodos:

  • __iter__() — deve retornar o próprio iterador (geralmente return self).
  • __next__() — devolve o próximo elemento da sequência, e ao chegar ao fim deve lançar uma exceção StopIteration.

Exemplo de um iterador simples:

class CountUpTo:
    def __init__(self, max):
        self.max = max
        self.current = 0

    def __iter__(self):
        return self

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

counter = CountUpTo(3)
for num in counter:
    print(num)  # Vai imprimir 1, 2, 3