默认为密钥的Python字典？

Question

有没有办法让defaultdict默认返回密钥？或者一些具有同等行为的数据结构？即,在初始化字典后d,

>>> d['a'] = 1
>>> d['a']
1
>>> d['b']
'b'
>>> d['c']
'c'

我只看到默认字典采用不带参数的函数,所以我不确定除了创建一种新的字典之外是否还有其他解决方案.

Answer 1

我会覆盖以下__missing__方法dict:

>>> class MyDefaultDict(dict):
...     def __missing__(self, key):
...         self[key] = key
...         return key
...
>>> d = MyDefaultDict()
>>> d['joe']
'joe'
>>> d
{'joe': 'joe'}

Answer 2

编辑: 哎呀,我刚刚意识到我文件中的代码最初来自另一个stackoverflow答案!/sf/answers/203871881/,请继续投票.

这就是我使用的 - 它是一个defaultdict变体,它将键作为参数传递给default-value工厂函数,该函数作为参数传递给init,而不是没有参数:

class keybased_defaultdict(defaultdict):
    def __missing__(self, key):
        if self.default_factory is None:
            raise KeyError(key)
        else:
            value = self[key] = self.default_factory(key)
            return value

这是你想要的用途:

>>> d = keybased_defaultdict(lambda x: x)
>>> d[1]
1
>>> d['a']
'a'

其他可能性:

>>> d = keybased_defaultdict(lambda x: len(x))
>>> d['a']
1
>>> d['abc']
3