Compreensão de dicionário

Chaves e o par chave: valor

A ideia é a mesma da compreensão de lista, com duas mudanças. Primeiro, você troca os colchetes por chaves, porque o resultado é um dicionário. Segundo, em vez de uma única expressão, você informa duas, separadas por dois-pontos: a da esquerda vira a chave e a da direita vira o valor. Todo o resto (o for, os filtros) funciona igual. Se você entendeu a compreensão de lista, esta é um passo pequeno.

numeros = [1, 2, 3, 4]

quadrados = {n: n * n for n in numeros}

print(quadrados)  # {1: 1, 2: 4, 3: 9, 4: 16}

Chaves { }, e o par n: n * n. A esquerda vira chave, a direita vira valor.

Casos úteis do dia a dia

A compreensão de dicionário resolve tarefas que apareciam o tempo todo com laços longos. Uma delas é montar um dicionário a partir de duas listas paralelas, uma com as chaves e outra com os valores, usando a função zip para casá-las item a item. Outra é indexar uma sequência, criando um dicionário que liga cada item à sua posição. Em ambos os casos, o que era um laço de várias linhas vira uma expressão que se lê de uma vez.

produtos = ["arroz", "feijao", "cafe"]
precos = [22.90, 8.50, 15.00]

tabela = {nome: preco for nome, preco in zip(produtos, precos)}

print(tabela)  # {'arroz': 22.9, 'feijao': 8.5, 'cafe': 15.0}

zip casa produtos com precos, e a compreensão monta o dicionário nome para preço.

Inverter um dicionário é outro clássico. Às vezes você tem nome para código e precisa de código para nome; basta percorrer os pares com items() e escrever o par ao contrário. Funciona bem quando os valores são únicos, porque chaves de dicionário não se repetem. Se dois valores forem iguais, um vai sobrescrever o outro, então é um caso a usar com consciência do que os dados contêm.

codigos = {"arroz": "A1", "feijao": "B2", "cafe": "C3"}

invertido = {codigo: nome for nome, codigo in codigos.items()}

print(invertido)  # {'A1': 'arroz', 'B2': 'feijao', 'C3': 'cafe'}

items() dá os pares; escrever codigo: nome inverte chaves e valores.

Filtrar pares com um if

Assim como na lista, você pode filtrar quais pares entram no dicionário novo com um if no fim. A condição pode olhar a chave, o valor ou os dois. Isso é ótimo para peneirar um dicionário grande e ficar só com o que interessa, sem criar um dicionário vazio e ir preenchendo com laço. O exemplo pega uma tabela de preços e monta outra só com os itens acima de dez reais, mantendo a estrutura chave para valor.

precos = {"arroz": 22.90, "feijao": 8.50, "cafe": 15.00, "sal": 3.00}

caros = {nome: preco for nome, preco in precos.items() if preco > 10}

print(caros)  # {'arroz': 22.9, 'cafe': 15.0}

O if no fim filtra: só os pares com preço acima de dez reais entram.