Python 的 defaultdict 本身不支持将键传入 default_factory，但可通过继承 UserDict 并重写 __missing__ 方法实现键感知的默认值生成，灵活支持任意键依赖逻辑。
python 的 `defaultdict` 本身不支持将键传入 `default_factory`，但可通过继承 `userdict` 并重写 `__missing__` 方法实现键感知的默认值生成，灵活支持任意键依赖逻辑。
defaultdict 的设计限制在于：其 default_factory 被调用时不接收任何参数，因此无法直接根据键（如 d[1] 中的 1）动态计算默认值。例如，defaultdict(lambda key: key * 2) 会报错或返回错误结果，因为该 lambda 实际从未被传入 key。
要实现“按需、按键生成默认值”的行为，推荐方案是自定义字典类，继承 collections.UserDict（比直接继承 dict 更安全、更符合封装规范），并重写特殊方法 __missing__ —— 该方法在访问不存在的键时自动触发，且明确接收 key 参数。
以下是一个简洁可靠的实现：
from collections import UserDict
from typing import Any, Callable, Hashable
class KeyedDefaultDict(UserDict):
def __init__(
self,
default_factory: Callable[[Hashable], Any],
*args: Any,
**kwargs: Any,
):
super().__init__(*args, **kwargs)
self.default_factory = default_factory
def __missing__(self, key: Hashable) -> Any:
value = self.default_factory(key)
self.data[key] = value # 自动缓存，避免重复计算
return value
# 使用示例
d = KeyedDefaultDict(lambda k: k * 2, {0: -1})
print(d[0]) # → -1 （来自初始数据，不触发 __missing__）
print(d[1]) # → 2 （触发 __missing__，k=1 → 1*2）
print(d[3]) # → 6 （k=3 → 3*2）
print(d) # → {0: -1, 1: 2, 3: 6}
✅ 优势说明：
- 完全兼容标准字典接口（支持
.keys(),.items(),in,update()等）； -
default_factory可为任意可调用对象（函数、lambda、类实例等），支持复杂逻辑（如lambda k: f"item_{k:04d}"或lambda k: [] if isinstance(k, str) else {}）； - 写入操作（
self.data[key] = value）确保默认值被持久化，后续访问直接命中，性能与defaultdict相当。
⚠️ 注意事项：
- 切勿在
__missing__中抛出异常（如KeyError），否则会中断正常流程；必须返回一个值； - 若需线程安全，需额外加锁（本实现为单线程友好）；
-
UserDict的self.data是底层存储字典，直接操作它比覆盖__setitem__更高效且不易出错。
通过这一模式，你不仅能复刻 defaultdict 的便利性，还能解锁基于键的智能默认值能力——无论是初始化计数器、懒加载配置、还是构建嵌套结构，都变得清晰可控。
Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！
