具有缺省键的Python词典

问题描述

我正在运行Python2.7.10。

我希望字典返回存储在特定键处的值，以防丢失项。例如，如下所示：

myD = dict(...)
return myD[key] if key in myD else myD[defaultKey]

为了清楚起见，我想调用myD[key]并在代码中返回正确的值，而不返回多余的if...else...

这不完全是defaultdict所做的(因为它接受一个函数作为默认调用)，也不完全是dict.setdefault()所做的，myD.get(key, ???)似乎也没有帮助。我可能应该从dict或defaultdict继承并重载__init__()和missing()方法，但我想不出一个好的方法来做到这一点。

---

解决方案

我不完全确定您想要什么(我没有阅读您问题下的所有备注)，但我认为这可能至少接近您想要的内容。

class DefaultKeyDict(dict):
    def __init__(self, default_key, *args, **kwargs):
        self.default_key = default_key
        super(DefaultKeyDict, self).__init__(*args, **kwargs)

    def __missing__ (self, key):
        if self.default_key not in self:  # default key not defined
            raise KeyError(key)
        return self[self.default_key]

    def __repr__(self):
        return ('{}({!r}, {})'.format(self.__class__.__name__,
                                      self.default_key,
                                      super(DefaultKeyDict, self).__repr__()))

    def __reduce__(self):  # optional, for pickle support
        args = (self.default_key if self.default_key in self else None,)
        return self.__class__, args, None, None, self.iteritems()

dkdict = DefaultKeyDict('b', {'a': 1, 'b': 2, 'c': 3})

print dkdict['a']  # -> 1
print dkdict['b']  # -> 2
print dkdict['c']  # -> 3
print dkdict['d']  # -> 2 (value of the default key)

del dkdict['b']  # delete the default key
try:
    print dkdict['d']  # should now raise exception like normal
except KeyError:
    print("dkdict['d'] raised a KeyError")

您可能希望修改类__init__()方法以接受默认键及其值作为参数(而不仅仅是键)。