Compreensões aninhadas

Dois for na mesma compreensão

Uma compreensão pode ter mais de um for, e isso corresponde a um laço dentro do outro. A pegadinha é a ordem: ela é a mesma que você escreveria nos laços normais, de fora para dentro. O primeiro for é o externo, o segundo é o interno. Muita gente inverte na cabeça porque em compreensões acha que se lê de trás para frente, mas não: a ordem dos for é a ordem natural dos laços.

# Laco aninhado tradicional:
pares = []
for letra in "AB":
    for numero in [1, 2, 3]:
        pares.append(f"{letra}{numero}")

print(pares)  # ['A1', 'A2', 'A3', 'B1', 'B2', 'B3']

O laço aninhado clássico: para cada letra, percorre todos os números.

# A mesma coisa como compreensao, os for na MESMA ordem:
pares = [f"{letra}{numero}" for letra in "AB" for numero in [1, 2, 3]]

print(pares)  # ['A1', 'A2', 'A3', 'B1', 'B2', 'B3']

for letra vem primeiro (laço de fora) e for numero depois (laço de dentro).

Achatar uma lista de listas

Um dos usos mais úteis do aninhamento é achatar uma lista de listas, transformando algo com dois níveis em uma sequência plana. Imagine uma matriz, que em Python costuma ser uma lista de linhas, cada linha sendo uma lista de números. Para juntar todos os números em uma lista só, você percorre cada linha e, dentro dela, cada item. Em compreensão, isso vira uma linha, com o for de fora pegando as linhas e o de dentro pegando os itens.

matriz = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

plana = [item for linha in matriz for item in linha]

print(plana)  # [1, 2, 3, 4, 5, 6, 7, 8, 9]

Achatar: for linha in matriz (de fora) e for item in linha (de dentro).

Há também o caso oposto e igualmente comum: montar uma matriz com uma compreensão dentro de outra. Aqui os colchetes aparecem duas vezes, os de fora criando a lista de linhas e os de dentro criando cada linha. Este é o ponto em que muita gente decide, com razão, que um for tradicional lê melhor. A versão em compreensão é elegante, mas exige leitura atenta, então use com bom senso.

# Tabuada 1 a 3, cada linha e uma lista:
tabuada = [[linha * col for col in range(1, 4)] for linha in range(1, 4)]

for linha in tabuada:
    print(linha)
# [1, 2, 3]
# [2, 4, 6]
# [3, 6, 9]

Compreensão dentro de compreensão cria uma matriz. Bonito, mas pede atenção ao ler.

Até onde o aninhamento vale a pena

Compreensão aninhada é o ponto do módulo em que o poder e o risco se cruzam. Com dois for ou uma compreensão dentro da outra, você economiza linhas, mas cada nível a mais cobra um preço de leitura. A pergunta a fazer nunca muda: quem ler isto daqui a três meses vai entender rápido? Se a resposta for não, os laços separados, com nomes claros e indentação, são a escolha profissional. Escrever menos não é o objetivo; escrever claro é.

Vale guardar que quase toda compreensão aninhada tem uma versão em laços comuns que faz exatamente a mesma coisa. Elas não competem; a compreensão é um atalho para os casos claros, e o for tradicional é o porto seguro para os casos densos. Um bom programador intermediário sabe escrever as duas e escolhe a que deixa a intenção mais óbvia naquele contexto. Essa escolha consciente é mais valiosa que decorar a sintaxe.