Generator expressions

Troque o colchete pelo parêntese

No módulo de compreensões você aprendeu a escrever [x * x for x in numeros] e transformar três linhas de laço em uma só. A generator expression é a irmã preguiçosa dessa sintaxe, e a mudança é quase invisível: troque os colchetes por parênteses. O resultado deixa de ser uma lista pronta e passa a ser um gerador, com todos os benefícios das aulas anteriores. Mesma legibilidade, mesma clareza de intenção, mas sem materializar a coleção inteira. É o jeito mais rápido e comum de criar um gerador no Python do dia a dia, sem precisar escrever uma função com yield.

numeros = [1, 2, 3, 4, 5]

lista = [n * n for n in numeros]     # compreensao de lista
gerador = (n * n for n in numeros)   # generator expression

print(lista)              # [1, 4, 9, 16, 25]
print(gerador)            # <generator object <genexpr> at 0x...>
print(list(gerador))      # [1, 4, 9, 16, 25]

Colchetes constroem a lista na hora; parênteses criam um gerador que produz sob demanda.

A diferença é a mesma da aula anterior: a versão com colchetes calcula os cinco quadrados de imediato e os guarda; a versão com parênteses cria um gerador que só calcula cada quadrado quando você o pede. Como o gerador se esgota depois de percorrido, imprimir gerador direto mostra apenas o objeto, e envolvê-lo em list() o consome de uma vez. Para cinco itens, prefira a lista, que é mais simples de reusar. O ganho aparece quando os itens são muitos, ou quando você só vai passar por eles uma vez.

Quando preferir cada uma

A regra prática é direta. Use a compreensão de lista quando você precisa da lista de verdade: para percorrer mais de uma vez, ordenar, contar o len, acessar por índice ou guardar para depois. Use a generator expression quando os dados serão consumidos uma única vez, em fluxo, especialmente se forem muitos. O caso mais elegante é passar a generator expression direto para uma função que consome um iterável, como sum, max, min, any ou all. Nesses casos você nem precisa dos parênteses extras, porque os da própria chamada da função bastam.

precos = [19.90, 5.50, 42.00, 8.75, 100.00]

# Soma sem criar lista intermediaria: a generator expression
# alimenta o sum() item por item.
total = sum(preco for preco in precos if preco > 10)
print(total)          # 161.9

# Maior preco acima de 10, tambem sem lista intermediaria.
maior = max(preco for preco in precos if preco > 10)
print(maior)          # 100.0

Passar a generator expression direto para sum e max evita construir uma lista só para descartá-la.

Um detalhe de leitura ajuda a escolher com confiança. Se você se pega escrevendo sum([x for x in dados]), quase sempre pode remover os colchetes de dentro e escrever sum(x for x in dados). A versão com colchetes constrói uma lista inteira só para o sum consumi-la e jogá-la fora em seguida, um desperdício de memória. A versão sem colchetes alimenta o sum diretamente, item por item. É uma daquelas trocas pequenas que deixam o código mais pythônico sem custo nenhum de clareza.

Fixando a escolha

A generator expression é a forma mais leve de trazer a preguiça dos geradores para o código do dia a dia. Ela tem a mesma cara da compreensão de lista, então não custa nada a mais para ler, e entrega a economia de memória quando os dados são muitos ou passageiros. Guardar a diferença de um caractere, colchete contra parêntese, e o que ela significa, é o que permite escolher a estrutura certa quase no automático. Na próxima aula você vai ver como conectar várias generator expressions e funções geradoras em sequência, formando pipelines de dados.