Início / coding / Perguntas / 79573449
Accepted
user30256092
Asked: 2025-04-14 23:01:04 +0800 CST

Removendo elementos com base em valores de dicionário aninhados

  • 772

Tenho uma estrutura complexa de dicionário aninhado e preciso remover elementos com base nos valores de um dicionário aninhado. Meu dicionário se parece com isto:

my_dict = {
    'item1': {'name': 'Apple', 'price': 1.0, 'category': {'id': 1, 'name': 'Fruit'}},
    'item2': {'name': 'Banana', 'price': 0.5, 'category': {'id': 1, 'name': 'Fruit'}},
    'item3': {'name': 'Carrot', 'price': 0.75, 'category': {'id': 2, 'name': 'Vegetable'}},
    'item4': {'name': 'Broccoli', 'price': 1.5, 'category': {'id': 2, 'name': 'Vegetable'}}
}

Quero filtrar este dicionário para incluir apenas itens pertencentes à categoria "Fruta". Tentei o seguinte código:

new_dict = {}
for key, value in my_dict.items():
    if value['category']['name'] == 'Fruit':
        new_dict[key] = value
print(new_dict)

Isso funciona, mas estou me perguntando se há uma maneira mais concisa ou Pythonica de fazer isso, talvez usando compreensão de dicionário ou uma função de filtragem como filter().

python

4 respostas

  1. Best Answer

    Como uma compreensão de dicionário

    Uma compreensão de dicionário pode ser usada para criar dicionários a partir de expressões arbitrárias de chave e valor .

    new_dict2 = {
    key: value
    for key, value in my_dict.items()
    if value['category']['name'] == 'Fruit'
}
new_dict2 == new_dict
# True

    Usandofilter()

    A filter()função é usada para:

    Construa um iterador a partir dos elementos do iterável para os quais a função é verdadeira.

    Retorna dict.items()um iterável onde cada elemento é uma tupla de comprimento 2. Podemos fornecer cada um itema uma função lambda , onde item[0]serão a chave e item[1]o valor. filter()Retorna um iterador das tuplas que correspondem à condição. Podemos encapsular isso dict()para obter um dicionário (da mesma forma que dict([("key1", "value1"), ("key2", "value2")])retorna {'key1': 'value1', 'key2': 'value2'}).

    new_dict3 = dict(
    filter(
        lambda item: item[1]['category']['name'] == 'Fruit',
        my_dict.items()
    )
)
new_dict3 == new_dict
# True

    A maneira mais pitônica

    Alcançar o objetivo nebuloso da Pythonicidade (Pythonicity?) é sempre um tanto subjetivo. Acho que a compreensão de um dicionário é limpa e organizada, mas pode ser difícil ver o que está acontecendo, especialmente se a função dictestiver profundamente aninhada ou se a condição for complexa. Provavelmente fica mais claro se você a envolver em uma função com nome apropriado para poder ver o que está acontecendo. Adicionei anotações de tipo para maior clareza:

    def find_fruit(d: dict[str, dict]) -> dict[str, dict]:

    def is_fruit(key: str, value: dict) -> bool:
        return value["category"]["name"] == "Fruit"

    return {key: value for key, value in d.items() if is_fruit(key, value)}

fruit_dict = find_fruit(my_dict)
new_dict == fruit_dict
# True

    Isso é basicamente o mesmo que a primeira abordagem, mas é mais agradável aos olhos.

    • 6

  2. Você poderia usar uma compreensão de dicionário que seria mais concisa, mas, sem dúvida, mais difícil de ler:

    my_dict = {
    'item1': {'name': 'Apple', 'price': 1.0, 'category': {'id': 1, 'name': 'Fruit'}},
    'item2': {'name': 'Banana', 'price': 0.5, 'category': {'id': 1, 'name': 'Fruit'}},
    'item3': {'name': 'Carrot', 'price': 0.75, 'category': {'id': 2, 'name': 'Vegetable'}},
    'item4': {'name': 'Broccoli', 'price': 1.5, 'category': {'id': 2, 'name': 'Vegetable'}}
}

new_dict = {k: v for (k, v) in my_dict.items() if v["category"]["name"] == "Fruit"}

print(new_dict)
    • 4

  3. Faça isso de uma maneira "pythônica":

    filtered_dict = {k: v for k, v in my_dict.items() if v['category']['name'] == 'Fruit'}
    • 3

  4. Você também pode fazer isso comfilter

    
new_dict = dict(filter(lambda item: item[1]['category']['name'] == 'Fruit', my_dict.items()))
    • 3

relate perguntas