defaultdict(list)是嵌套分组首选,自动初始化空列表避免键存在性检查;两层用defaultdict(lambda: defaultdict(list));三层及以上推荐setdefault或封装函数;键必须不可变。

defaultdict(list) 是最常用的嵌套分组起点
多数嵌套分组需求本质是「按某键聚合多个值」,比如把 [('A', 1), ('B', 2), ('A', 3)] 变成 {'A': [1, 3], 'B': [2]}。直接用 dict 需反复检查键是否存在,而 defaultdict(list) 自动初始化空列表,省掉 if key not in d: d[key] = [] 这类样板。
注意:不要写成 defaultdict([]) —— 这会把空列表对象作为默认工厂,所有键共享同一份引用,导致数据污染。必须传可调用对象,即 list(类型名),不是 [](实例)。
常见错误现象:defaultdict(list) 后往某个键 append 数据,结果所有键的列表都变长了 —— 就是踩了上面那个坑。
两层嵌套用 defaultdict(lambda: defaultdict(list))
当需要按两个维度分组,比如「先按城市,再按部门统计员工姓名」,结构类似 {'北京': {'研发': ['张三', '李四'], '市场': ['王五']}, '上海': {...}},这时得嵌套工厂函数。
推荐写法:defaultdict(lambda: defaultdict(list))。不能写 defaultdict(defaultdict(list)),因为后者在构造时就执行了内层 defaultdict(list),返回的是一个实例,不是可调用对象,会报 TypeError: first argument must be callable。
使用场景示例:
from collections import defaultdict
data = [('北京', '研发', '张三'), ('北京', '研发', '李四'), ('上海', '市场', '王五')]
grouped = defaultdict(lambda: defaultdict(list))
for city, dept, name in data:
grouped[city][dept].append(name)
# grouped['北京']['研发'] → ['张三', '李四']
三层及以上建议封装成函数或用 setdefault
继续套 lambda: defaultdict(lambda: defaultdict(list)) 虽然语法可行,但可读性陡降,且容易漏掉某层的 lambda 包裹。更稳妥的做法是封装一个生成嵌套 defaultdict 的工具函数,或者改用 dict.setdefault() 配合普通 dict。
setdefault 在键不存在时设置默认值并返回,存在则直接返回原值,语义清晰、无嵌套陷阱:
-
d.setdefault('A', {})['B'] = 1等价于「确保 d['A'] 是 dict,然后赋值」 - 比多层 defaultdict 更易调试,出错时堆栈指向明确行号
- 对性能影响微乎其微,现代 CPython 中
setdefault是原子操作
如果坚持用 defaultdict,三层推荐:defaultdict(lambda: defaultdict(lambda: defaultdict(list))) —— 每一层 lambda 都要独立包裹,少一个括号或 lambda 就会触发默认值复用或类型错误。
注意 key 类型和不可变性限制
defaultdict 的键仍受 Python 字典约束:必须是不可变类型。常见踩坑点是误用 list 或 dict 做键,比如想用 grouped[[1, 2, 3]].append(x),会立刻抛 TypeError: unhashable type: 'list'。
解决方法:
- 元组替代列表:
grouped[(1, 2, 3)].append(x) - 字符串拼接(仅限简单场景):
grouped[f"{a}_{b}_{c}"].append(x) - 对复杂对象,提前定义
__hash__和__eq__,但需确保逻辑正确且 hash 不变
另外,defaultdict 不会自动递归处理嵌套结构中的缺失键 —— 它只在最外层键缺失时触发默认工厂。深层访问如 d['a']['b']['c'].append(1) 中,若 'a' 存在但 'b' 不存在,仍会报 KeyError,除非你已为 d['a'] 赋值成另一个 defaultdict。
Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!











