16

collections.defaultdict是很棒的。特别是与lambda：

>>> import collections
>>> a = collections.defaultdict(lambda : [None,None])
>>> a['foo']
[None, None]

有没有办法'foo'在 lambda 中使用给定的键（例如）？例如（不起作用）：

>>> a = collections.defaultdict(lambda : [None]*key)
>>> a[1]
[None]
>>> a[2]
[None, None]
>>> a
defaultdict(<function <lambda> at 0x02984170>, {1: [None], 2: [None, None]})
4

3 回答 3

32

您可能希望每当您尝试访问字典中不存在的项目时都会__missing__调用它；dict香草__missing__引发了一个例外，但你可以在一个子类中做任何你喜欢的事情：

class A(dict):
    def __missing__(self, key):
        value = self[key] = [None] * key
        return value
于 2011-11-01T12:10:44.067 回答
4

结合SingleNegationElimination和rplnt的答案以及defaultdict 文档，我使用了以下解决方案。

import collections
class KeyBasedDefaultDict(collections.defaultdict):
    def __missing__(self, key):
        if self.default_factory is None:
            raise KeyError(key)
        self[key] = self.default_factory(key)
        return self[key]

该方法的主体可能只是return self.default_factory(key)，但额外的代码确保复制所有 defaultdict 行为。

问题描述的用法：

d = KeyBasedDefaultDict(lambda key: [None] * key)
d[1]
> [None]
d[2]
> [None, None]
于 2017-06-23T23:30:10.993 回答
1

这将按要求工作，尽管它可能不是最好的解决方案（您需要使用默认调用对其进行初始化，然后不使用它）。它可能可以通过覆盖其他一些方法来修复。

class NoneDict(collections.defaultdict):
    def __setitem__(self, key, value):
        super(NoneDict, self).__setitem__(key, key*[None])
于 2011-11-01T08:10:48.897 回答