实现有序的默认字典
Python 中的内置集合模块提供 OrderedDict 和 defaultdict,每个都提供不同的功能。 OrderedDict 保留其元素的插入顺序,而 defaultdict 根据指定的工厂函数自动生成缺失键的默认值。
结合两者的力量
结合这两种数据结构的功能,我们可以创建一个有序的默认字典,该字典保留其元素的顺序并使用默认值初始化缺失的键。这可以通过扩展 OrderedDict 类来实现。
自定义类:DefaultOrderedDict
以下自定义类 DefaultOrderedDict 继承自 OrderedDict 并添加处理缺失键的功能:
<code class="python">from collections import OrderedDict, Callable class DefaultOrderedDict(OrderedDict): def __init__(self, default_factory=None, *a, **kw): if (default_factory is not None and not isinstance(default_factory, Callable)): raise TypeError('first argument must be callable') OrderedDict.__init__(self, *a, **kw) self.default_factory = default_factory def __getitem__(self, key): try: return OrderedDict.__getitem__(self, key) except KeyError: return self.__missing__(key) def __missing__(self, key): if self.default_factory is None: raise KeyError(key) self[key] = value = self.default_factory() return value</code>
用法
这个类可以像普通的 OrderedDict 一样使用:
<code class="python">ordered_default_dict = DefaultOrderedDict(lambda: 0) ordered_default_dict['key1'] = 1 ordered_default_dict['key2'] = 2</code>
但是,如果一个键是不存在,则生成工厂函数的默认值:
<code class="python">print(ordered_default_dict['missing_key']) # Output: 0</code>
以上是如何在 Python 中创建有序的默认字典?的详细内容。更多信息请关注PHP中文网其他相关文章!