defaultdict 基本用法

lyj

发布 2025-01-20

255

分享

> `defaultdict` 是 Python 的 `collections` 模块中的一个工具，它是 `dict` 的子类，但与普通的字典不同，它在访问不存在的键时会自动为该键生成一个默认值，而不是抛出 `KeyError`。 ### defaultdict 基本用法 1. **导入和创建** ```python from collections import defaultdict # 创建一个 defaultdict，默认值为 int（即 0） dd = defaultdict(int) ``` 2. **使用** - 当访问一个不存在的键时，defaultdict 会使用提供的工厂函数生成一个默认值，并将其赋给该键。 ```python dd['a'] += 1 # 'a' 的默认值是 0，之后变为 1 print(dd) # defaultdict(<class 'int'>, {'a': 1}) ``` ### 常见用法 1. **计数器** `defaultdict` 可以用来替代 `Counter` 进行简单的计数操作： ```python text = "apple banana apple orange banana apple" word_count = defaultdict(int) for word in text.split(): word_count[word] += 1 print(word_count) # defaultdict(<class 'int'>, {'apple': 3, 'banana': 2, 'orange': 1}) ``` 2. **分组** 将数据按照某个条件分组： ```python data = [('a', 1), ('b', 2), ('a', 3), ('b', 4), ('c', 5)] group = defaultdict(list) for key, value in data: group[key].append(value) print(group) # defaultdict(<class 'list'>, {'a': [1, 3], 'b': [2, 4], 'c': [5]}) ``` 3. **嵌套字典** 方便处理嵌套结构的字典： ```python nested_dict = defaultdict(lambda: defaultdict(int)) nested_dict['a']['b'] += 1 print(nested_dict) # defaultdict(<function <lambda>>, {'a': defaultdict(<class 'int'>, {'b': 1})}) ``` 4. **默认值处理** 创建一个默认返回固定值的字典： ```python dd = defaultdict(lambda: 'default_value') print(dd['unknown']) # default_value ``` ### 使用注意事项 1. **工厂函数** - 工厂函数必须是一个可调用对象，如类型（`int`, `list`, 等）或自定义函数（`lambda`）。 - 如果不提供工厂函数，`defaultdict` 会报错。 2. **默认值的插入** 访问不存在的键时，`defaultdict` 会自动创建一个键并插入默认值，这一点可能导致意外行为： 2. **避免误用** - 如果没有特别需要默认值的功能，普通字典可能更合适。