python - 有没有巧妙的方法将 key 传递给 defaultdict 的 default_factory？

Question

一个类有一个带有一个参数的构造函数:

class C(object):
    def __init__(self, v):
        self.v = v
        ...

在代码中的某个地方，dict 中的值知道它们的键是很有用的。
我想使用带有传递给新生儿默认值的键的 defaultdict:

d = defaultdict(lambda : C(here_i_wish_the_key_to_be))

有什么建议吗？

Answer 1

它几乎不能称得上是 聪明 - 但子类化是你的 friend :

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

d = keydefaultdict(C)
d[x] # returns C(x)