itertools

itertools fornece blocos de construção de iteradores rápidos e eficientes em memória - encadeamento, fatiamento, combinatória e agrupamento sem listas intermediárias.

Receita

from itertools import chain, islice, product
 
flat = chain([1, 2], [3, 4])
first_three = list(islice(range(10), 3))
pairs = list(product(["A", "B"], [1, 2]))

Quando usar:

• Achatar iteráveis aninhados
• Paginação via islice em um contador infinito
• Agrupar linhas de log ordenadas por chave
• Produtos cartesianos para geração de casos de teste

Exemplo de Trabalho

from itertools import chain, groupby, islice, accumulate
 
def group_sorted(rows: list[tuple[str, int]]):
    for key, group in groupby(rows, key=lambda r: r[0]):
        yield key, sum(v for _, v in group)
 
def paginate(iterator, page_size: int):
    page = 0
    while True:
        chunk = list(islice(iterator, page_size))
        if not chunk:
            break
        yield page, chunk
        page += 1
 
def running_totals(values):
    return accumulate(values)
 
if __name__ == "__main__":
    rows = [("a", 1), ("a", 2), ("b", 3)]
    print(list(group_sorted(sorted(rows))))
    print(list(paginate(iter(range(7)), 3)))
    print(list(running_totals([1, 2, 3, 4])))

O que isso demonstra:

• groupby precisa ser ordenado pela mesma função de chave primeiro
• islice pega de qualquer iterador sem len()
• Padrão paginate para APIs de streaming
• accumulate produz reduções correntes (soma padrão)

Mergulho Profundo

Como Funciona

• Álgebra de iteradores - Funções consomem iteráveis, retornam iteradores.
• groupby - Agrupa apenas chaves consecutivas iguais - não é SQL GROUP BY.
• tee - Divide o iterador em iteradores independentes com armazenamento em buffer.
• cycle/repeat/count - Iteradores infinitos - sempre limite o loop do consumidor.
• combinations/permutations - Geração combinatória preguiçosa.

Frequentemente Usado

Notas do Python

from itertools import pairwise  # 3.10+
 
for a, b in pairwise([1, 2, 3]):
    print(a, b)

Armadilhas

• groupby sem sort - Divide grupos incorretamente quando as chaves não são contíguas. Correção: sorted(data, key=key) primeiro.
• tee memory - Armazena valores em buffer para iteradores secundários. Correção: Use com moderação em fluxos grandes.
• produto infinito - product(count(), repeat=2) nunca termina. Correção: Limite com islice.
• chain.from_iterable - Achata um nível - aninhamento mais profundo requer recursão ou chain aninhado.
• zip vs zip_longest - zip para no mais curto; zip_longest precisa de fillvalue para preenchimento.

Alternativas

FAQs

Relacionado

• Geradores e yield - iteradores customizados
• functools - reduce relacionado a accumulate
• Compreensões e Expressões Geradoras - genexp
• Módulo collections - Counter com iteráveis

Versões do Stack: Esta página foi escrita para Python 3.14.0 (estável 3.14, manutenção 3.13), FastAPI 0.115+, Django 5.2, Flask 3.1, Pydantic 2, PyTorch 2.6+, pandas 2.2+, Polars 1.x, ruff 0.9+, e uv 0.6+.