如何在 Python 中创建有序的默认字典？

实现有序的默认字典

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中文网其他相关文章！