如何将 collections.defaultdict 转换为普通字典

简介

在 Python 编程领域，collections 模块提供了一个名为 defaultdict 的强大工具，它可以简化字典中缺失键的处理。然而，有时你可能需要将 defaultdict 转换为普通字典。本教程将指导你完成将 collections.defaultdict 转换为普通字典的过程，并探讨此技术的实际用例。

理解 collections.defaultdict

什么是 collections.defaultdict？

collections.defaultdict 是 Python 内置 dict 类的一个子类。它提供了一种创建类似字典对象的方式，该对象对于缺失的键有一个默认值。这意味着当你尝试访问字典中不存在的键时，defaultdict 不会引发 KeyError，而是会自动使用默认值创建一个新的条目。

defaultdict 中的默认值

defaultdict 的默认值在创建对象时指定。这可以是任何可调用对象，例如函数、类或 lambda 表达式。当访问一个新键时，返回默认值，并将键值对添加到字典中。

下面是一个示例：

from collections import defaultdict

## 创建一个默认值为 0 的 defaultdict
dd = defaultdict(int)
dd['a'] = 1
dd['b'] += 1
print(dd)  ## 输出：defaultdict(<class 'int'>, {'a': 1, 'b': 1})
print(dd['c'])  ## 输出：0

在这个示例中，当我们尝试访问字典中不存在的键 'c' 时，defaultdict 会自动创建一个默认值为 0（int 类型的默认值）的新条目。

实际用例

defaultdict 在各种场景中都很有用，例如：

计数出现次数：你可以使用 defaultdict(int) 轻松统计列表或其他可迭代对象中元素的出现次数。
分组数据：你可以使用 defaultdict(list) 按某个键对数据进行分组，其中值存储在列表中。
嵌套字典：你可以使用 defaultdict(dict) 创建嵌套字典，而无需检查中间键是否存在。

通过理解 collections.defaultdict 的基础知识，你可以利用它的便利性和灵活性来简化你的 Python 代码，并更优雅地处理缺失的键。

将 defaultdict 转换为普通字典

为什么要将 defaultdict 转换为普通字典？

虽然 defaultdict 是一个强大且方便的工具，但在某些情况下，你可能需要将其转换回普通的 dict。如果你需要将字典传递给期望标准 dict 对象的函数或库，或者如果你想执行一些 defaultdict 不支持的操作，这可能是必要的。

将 defaultdict 转换为 dict 的方法

有几种方法可以将 defaultdict 转换为普通的 dict：

使用 dict() 构造函数：

from collections import defaultdict

dd = defaultdict(int)
dd['a'] = 1
dd['b'] = 2

regular_dict = dict(dd)
print(regular_dict)  ## 输出：{'a': 1, 'b': 2}

遍历 defaultdict 并创建一个新的 dict：

from collections import defaultdict

dd = defaultdict(int)
dd['a'] = 1
dd['b'] = 2

regular_dict = {k: v for k, v in dd.items()}
print(regular_dict)  ## 输出：{'a': 1, 'b': 2}

使用 copy() 方法：

from collections import defaultdict

dd = defaultdict(int)
dd['a'] = 1
dd['b'] = 2

regular_dict = dd.copy()
print(regular_dict)  ## 输出：{'a': 1, 'b': 2}

这些方法确保生成的 dict 对象具有与原始 defaultdict 相同的键值对，但没有默认值功能。

转换时的注意事项

将 defaultdict 转换为普通 dict 时，请记住你将失去默认值行为。如果你尝试访问生成的 dict 中不存在的键，你将得到一个 KeyError，而不是默认值。

通过了解如何将 defaultdict 转换为普通 dict，你可以无缝地将 defaultdict 集成到你的 Python 代码中，并处理需要标准 dict 的情况。

实际用例

计数出现次数

defaultdict 的一个常见用例是统计列表或其他可迭代对象中元素的出现次数。通过使用 defaultdict(int)，你可以轻松跟踪每个元素的计数，而无需检查键是否已存在于字典中。

from collections import defaultdict

words = ['apple', 'banana', 'cherry', 'apple', 'banana', 'date']
word_count = defaultdict(int)

for word in words:
    word_count[word] += 1

print(dict(word_count))
## 输出：{'apple': 2, 'banana': 2, 'cherry': 1, 'date': 1}

分组数据

defaultdict 的另一个有用应用是按某个键对数据进行分组，其中值存储在列表中。当你需要根据某些标准组织数据时，这可能特别有帮助。

from collections import defaultdict

data = [
    {'name': 'Alice', 'age': 25, 'city': 'New York'},
    {'name': 'Bob', 'age': 30, 'city': 'Los Angeles'},
    {'name': 'Charlie', 'age': 35, 'city': 'New York'},
    {'name': 'David', 'age': 40, 'city': 'Los Angeles'},
]

grouped_data = defaultdict(list)
for item in data:
    grouped_data[item['city']].append(item)

print(dict(grouped_data))
## 输出：{'New York': [{'name': 'Alice', 'age': 25, 'city': 'New York'}, {'name': 'Charlie', 'age': 35, 'city': 'New York'}],
##          'Los Angeles': [{'name': 'Bob', 'age': 30, 'city': 'Los Angeles'}, {'name': 'David', 'age': 40, 'city': 'Los Angeles'}]}

嵌套字典

在处理嵌套字典时，defaultdict 也很有用。通过使用 defaultdict(dict)，当访问不存在的键时，你可以自动创建新的嵌套字典。

from collections import defaultdict

data = {
    'fruit': {
        'apple': 2,
        'banana': 3,
    },
   'vegetable': {
        'carrot': 5,
        'broccoli': 4,
    },
}

nested_dd = defaultdict(dict)
for category, items in data.items():
    for item, count in items.items():
        nested_dd[category][item] = count

print(dict(nested_dd))
## 输出：{'fruit': {'apple': 2, 'banana': 3},'vegetable': {'carrot': 5, 'broccoli': 4}}

通过探索这些实际用例，你可以看到 defaultdict 如何简化你的 Python 代码，并帮助你更有效地处理缺失的键。

总结

在本教程结束时，你将对如何将 Python 的 collections.defaultdict 转换为普通字典有扎实的理解。你将了解这两种数据结构之间的关键区别，并发现这种转换在哪些实际用例中会有所帮助。有了这些知识，你将能够更高效地使用 Python 的内置数据结构，并提升你的整体编程技能。