主页 / coding / 问题 / 79573449
Accepted
user30256092
Asked: 2025-04-14 23:01:04 +0800 CST

根据嵌套字典值删除元素

  • 772

我有一个复杂的嵌套字典结构，需要根据嵌套字典中的值删除元素。我的字典如下所示：

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 = {}
for key, value in my_dict.items():
    if value['category']['name'] == 'Fruit':
        new_dict[key] = value
print(new_dict)

这是可行的，但我想知道是否有更简洁或更 Pythonic 的方法来实现这一点，也许使用字典理解或类似的过滤函数filter()。

python

4 个回答

  1. Best Answer

    作为字典理解

    字典推导式 可用于根据任意键和值表达式创建字典。

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

    使用filter()

    该filter()函数用于：

    从可迭代的函数为真的元素构造一个迭代器。

    返回dict.items()一个可迭代对象，其中每个元素都是一个长度为 2 的元组item。我们可以将每个元素传入一个lambda 函数，其中item[0]是键， 是item[1]值。filter()返回一个符合条件的元组的迭代器。我们可以将其包装进去得到一个字典（与返回 的dict()方法相同）。dict([("key1", "value1"), ("key2", "value2")]){'key1': 'value1', 'key2': 'value2'}

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

    最 Pythonic 的方式

    实现“Pythonicness”（Pythonicity？）这个模糊的目标总是有些主观。我认为字典推导式简洁明了，但很难看清它在做什么，尤其是在dict嵌套很深或条件很复杂的情况下。最好把它包装在一个命名合适的函数中，这样就能看清到底发生了什么。为了清晰起见，我添加了类型注解：

    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

    这与第一种方法基本相同，但更容易看清。

    • 6

  2. 您可以使用字典推导式，它会更简洁，但可以说更难阅读：

    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. 以“pythonic”方式执行此操作：

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

  4. 你也可以这样做filter

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

相关问题