Introdução aos iteradores e iteráveis

Iteradores e iteráveis são conceitos fundamentais no Python que permitem trabalhar com sequências de dados. Eles são a base dos laços, das compreensões e dos geradores.

Nesta lição, você aprenderá como os iteradores e iteráveis funcionam e como criar seus próprios iteradores.

O que é um iterável?

Um iterável é qualquer objeto no Python que pode ser percorrido usando um laço for. Exemplos incluem listas, tuplas, conjuntos e dicionários.

# Example of an iterable
my_list = [1, 2, 3]
for item in my_list:
    print(item)  # Outputs: 1, 2, 3

O que é um iterador?

Um iterador é um objeto que representa um fluxo de dados. Ele implementa os métodos __iter__() e __next__().

# Example of an iterator
my_list = [1, 2, 3]
iterator = iter(my_list)
print(next(iterator))  # Outputs: 1
print(next(iterator))  # Outputs: 2
print(next(iterator))  # Outputs: 3

A diferença entre iteráveis e iteradores

Os iteráveis podem ser percorridos (por exemplo, listas), enquanto os iteradores produzem um item por vez usando o método next(). Um iterável se torna um iterador quando é passado para a função iter().

Criando iteradores personalizados

Você pode criar iteradores personalizados definindo uma classe com os métodos __iter__() e __next__():

# Custom iterator example
class Counter:
    def __init__(self, start, end):
        self.current = start
        self.end = end

    def __iter__(self):
        return self

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

counter = Counter(1, 5)
for number in counter:
    print(number)  # Outputs: 1, 2, 3, 4, 5

Usando a função iter()

A função iter() retorna um objeto iterador a partir de um iterável:

# Using iter() on a list
my_list = [1, 2, 3]
iterator = iter(my_list)
print(next(iterator))  # Outputs: 1

Tratando StopIteration

Quando um iterador se esgota, ele gera uma exceção StopIteration. É assim que o Python sinaliza o fim de uma sequência.

# Handling StopIteration
my_list = [1, 2, 3]
iterator = iter(my_list)

while True:
    try:
        print(next(iterator))
    except StopIteration:
        break

Iteradores versus geradores

Os geradores são uma maneira mais simples de criar iteradores. Eles implementam automaticamente os métodos __iter__() e __next__().

# Example of a generator as an iterator
def my_generator():
    yield 1
    yield 2
    yield 3

for value in my_generator():
    print(value)  # Outputs: 1, 2, 3

Erros comuns com iteradores

Veja alguns erros que você deve evitar:

• Esquecer de implementar __iter__() e __next__() em iteradores personalizados.
• Não tratar corretamente as exceções StopIteration.
• Confundir iteradores com iteráveis.

O que aprendemos?

Nesta lição, você aprendeu:

• A diferença entre iteráveis e iteradores.
• Como usar __iter__() e __next__() para criar iteradores personalizados.
• Como usar a função iter() e tratar exceções StopIteration.
• Como os geradores simplificam a criação de iteradores.

Muito bem! Vamos passar ao próximo tópico.