Como posso usar if / else na compreensão de um dicionário?

138

Existe uma maneira no Python 2.7+ para criar algo como o seguinte?

{ something_if_true if condition else something_if_false for key, value in dict_.items() }

Eu sei que você pode fazer qualquer coisa com apenas 'se':

{ something_if_true for key, value in dict_.items() if condition}
diegueus9
fonte
4
como dito por @Marcin, a dicté feito de key:valueelementos, você não está construindo um dictaqui, mas um set(veja o conjunto de literais ).
Mdeous

Respostas:

247

Você já entendeu: A if test else Bé uma expressão Python válida. O único problema com sua compreensão de ditado, como mostrado, é que o local para uma expressão em uma compreensão de ditado deve ter duas expressões, separadas por dois pontos:

{ (some_key if condition else default_key):(something_if_true if condition
          else something_if_false) for key, value in dict_.items() }

A ifcláusula final atua como um filtro, diferente de ter a expressão condicional.

Marcin
fonte
28
Vale ressaltar que você não precisa ter uma condição if-else para a chave e o valor. Por exemplo, {(a if condition else b): value for key, value in dict.items()}vai funcionar.
Jeremy Weirich 26/07
5
@ JeremyWeirich Você não precisa ter um if-else para nenhum deles, se não quiser.
26616 Marcin
@ Marcin É possível usar apenas "if" para a parte chave e usar "if" e "else" para a parte do valor?
nithin11 12/06
14

A resposta de @ Marcin cobre tudo, mas, caso alguém queira ver um exemplo real, adiciono dois abaixo:

Digamos que você tenha o seguinte dicionário de conjuntos

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

e você deseja criar um novo dicionário cujas chaves indiquem se a sequência 'a'está contida nos valores ou não, você pode usar

dout = {"a_in_values_of_{}".format(k) if 'a' in v else "a_not_in_values_of_{}".format(k): v for k, v in d.items()}

que produz

{'a_in_values_of_key1': {'a', 'b', 'c'},
 'a_not_in_values_of_key2': {'bar', 'foo'},
 'a_not_in_values_of_key3': {'sad', 'so'}}

Agora vamos supor que você tenha dois dicionários como este

d1 = {'bad_key1': {'a', 'b', 'c'}, 'bad_key2': {'foo', 'bar'}, 'bad_key3': {'so', 'sad'}}
d2 = {'good_key1': {'foo', 'bar', 'xyz'}, 'good_key2': {'a', 'b', 'c'}}

e você deseja substituir as chaves d1pelas chaves, d2se os respectivos valores forem idênticos, você poderá

# here we assume that the values in d2 are unique
# Python 2
dout2 = {d2.keys()[d2.values().index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

# Python 3
dout2 = {list(d2.keys())[list(d2.values()).index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

que dá

{'bad_key2': {'bar', 'foo'},
 'bad_key3': {'sad', 'so'},
 'good_key2': {'a', 'b', 'c'}}
Cleb
fonte
para o seu segundo exemplo usando d1, d2eu receboAttributeError: 'dict_values' object has no attribute 'index'
alancalvitti
@alancalvitti: obrigado por apontar isso! A solução foi para o Python 2 e não funciona para o Python 3; Também adicionei uma solução Python 3.
Cleb 23/02/19
3

Caso você tenha condições diferentes para avaliar chaves e valores, a resposta de @ Marcin é o caminho a seguir.

Se você tiver a mesma condição para chaves e valores, é melhor criar túneis (chave, valor) em uma expressão de gerador alimentando dict():

dict((modify_k(k), modify_v(v)) if condition else (k, v) for k, v in dct.items())

É mais fácil ler e a condição é avaliada apenas uma vez por chave, valor.

Exemplo com o empréstimo do dicionário de conjuntos do @ Cleb:

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

Suponha que você deseje o sufixo apenas keyscom ano seu valuee que você queira valuesubstituir o comprimento do conjunto nesse caso. Caso contrário, o par de valores-chave deve permanecer inalterado.

dict((f"{k}_a", len(v)) if "a" in v else (k, v) for k, v in d.items())
# {'key1_a': 3, 'key2': {'bar', 'foo'}, 'key3': {'sad', 'so'}}
Darkonaut
fonte
0

Outro exemplo no uso de if / else na compreensão do dicionário

Estou trabalhando no aplicativo de desktop de entrada de dados para o meu próprio trabalho de escritório, e é comum que esse aplicativo de entrada de dados obtenha todas as entradas do widget de entrada e as despeje em um dicionário para processamento adicional, como validação ou edição, que devemos retornar dados selecionados do arquivo de volta aos widgets de entrada etc.

A primeira rodada usando codificação tradicional (8 linhas):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic, b_dic = {}, {}

for field, value in entries.items():
    if field == 'ther':
        for k,v in value.items():
            b_dic[k] = v
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

Na segunda rodada, tentei usar a compreensão do dicionário, mas o loop ainda está lá (6 linhas):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

for field, value in entries.items():
    if field == 'ther':
        b_dic = {k:v for k,v in value.items()}
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

Por fim, com uma declaração de compreensão de dicionário de uma linha (1 linha):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic = {field:{k:v for k,v in value.items()} if field == 'ther' 
        else value for field, value in entries.items()}
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

Eu uso python 3.8.3

KokoEfraim
fonte