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 (geralmentereturn self).__next__()— devolve o próximo elemento da sequência, e ao chegar ao fim deve lançar uma exceçãoStopIteration.
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