如何在 Python 中实现有序的 defaultdict 来维护键顺序并分配默认值?

Mary-Kate Olsen
发布: 2024-10-28 09:33:02
原创
521 人浏览过

How do you implement an ordered defaultdict in Python to maintain key order and assign default values?

实现有序默认字典

人们可能希望将 Python Collections 模块中的 OrderedDict() 和 defaultdict() 的功能合并到创建一个有序的默认字典。这种组合将允许维护一个字典,其中键是有序的,并且在访问不存在的键时分配默认值。

为了实现这一点,我们可以利用 Stack Overflow 社区中提供的配方的修改版本。代码如下:

<code class="python">from collections import OrderedDict, Callable

class DefaultOrderedDict(OrderedDict):
    # Source: http://stackoverflow.com/a/6190500/562769
    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

    # Override necessary methods for pickling
    def __reduce__(self):
        if self.default_factory is None:
            args = tuple()
        else:
            args = self.default_factory,
        return type(self), args, None, None, self.items()

    def copy(self):
        return self.__copy__()

    def __copy__(self):
        return type(self)(self.default_factory, self)

    def __deepcopy__(self, memo):
        import copy
        return type(self)(self.default_factory,
                          copy.deepcopy(self.items()))

    def __repr__(self):
        return 'OrderedDefaultDict(%s, %s)' % (self.default_factory,
                                               OrderedDict.__repr__(self))</code>
登录后复制

此代码片段包含一个 __init__ 方法,允许您指定默认工厂函数。 __missing__ 方法处理按键访问并使用工厂函数分配默认值。此外,还定义了 __reduce__、copy()、__copy__()、__deepcopy__() 和 __repr__() 等各种方法,用于正确腌制和表示此自定义字典。

通过使用此 DefaultOrderedDict 类,您可以构造一个有序字典,为缺失的键提供默认值。这种方法结合了有序密钥管理和动态值分配的优点。

以上是如何在 Python 中实现有序的 defaultdict 来维护键顺序并分配默认值?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责声明 Sitemap
PHP中文网:公益在线PHP培训,帮助PHP学习者快速成长!