时间戳

首页 > 解决方案 > 具有基于键的默认值的 Python dict

python - 具有基于键的默认值的 Python dict

问题描述

我需要一个字典，该字典会自动填充每个缺少的访问键的默认值。我已经找到defaultdict了一些其他方法来实现这一点，但我的问题是我希望每个键的默认值特定于键本身。

例如，defaultdict我可以实现这样的目标：

from collections import defaultdict
d = defaultdict(lambda: 5)
> d[1] = 3
> d[1]
> 3
> d[2]
> 5

但是，如果我需要例如每个访问的缺失键的默认值key + 5呢？就像是：

from collections import defaultdict
d = defaultdict(lambda key: key + 5)  # <-- This does not work as defaultdict expects lambda function to be without any parameters
> d[1] = 3
> d[1]
> 3
> d[2]
> 7         <- Calculated from accessed key + 5 (2+5)
> d[5]
> 10        <- Calculated from accessed key + 5 (5+5)

是否有一种干净的内置方式来实现我的需要？我知道我可以创建 dict 的子类并在__getitem__级别上实现这个特定的功能，但如果可能的话我想避免这种情况。

我在其他答案中找不到解决方案，如果它仍然是重复的，很抱歉。

标签： pythonpython-3.xdictionarydefaultdict

解决方案

我认为没有内置的方法可以做到这一点。但是，您可以对自身进行子类化，而不是子类化dict和 change ，以告诉使用参数 (the )调用，而不是不使用 args。它仍然是一个自定义类，而不是内置类，但它应该仍然非常有效。getitemdefaultdict__missing__()default_factorykey

from collections import defaultdict

class DefaultDict(defaultdict):
    def __missing__(self, key):
        return self.default_factory(key)

然后：

d = DefaultDict(lambda key: key + 5)

d[2]
# 7

推荐阅读