Python:defaultdict的defaultdict？

Question

有没有办法defaultdict(defaultdict(int))让以下代码工作？

for x in stuff:
    d[x.a][x.b] += x.c_int

d需要根据x.a和x.b元素进行临时构建.

我可以用:

for x in stuff:
    d[x.a,x.b] += x.c_int

但后来我无法使用:

d.keys()
d[x.a].keys()

Answer 1

是这样的:

defaultdict(lambda: defaultdict(int))

当您尝试访问不存在的密钥时,将调用a defaultdict(在本例中为lambda: defaultdict(int))的参数.它的返回值将被设置为此键的新值,这意味着在我们的情况下d[Key_doesnt_exist]将是值defaultdict(int).

如果您尝试从最后一个defaultdict访问一个键,即d[Key_doesnt_exist][Key_doesnt_exist]它将返回0,这是最后一个defaultdict的参数的返回值,即int().

Answer 2

defaultdict构造函数的参数是将用于构建新元素的函数.所以让我们用一个lambda!

>>> from collections import defaultdict
>>> d = defaultdict(lambda : defaultdict(int))
>>> print d[0]
defaultdict(<type 'int'>, {})
>>> print d[0]["x"]
0

从Python 2.7开始,使用Counter有一个更好的解决方案:

>>> from collections import Counter
>>> c = Counter()
>>> c["goodbye"]+=1
>>> c["and thank you"]=42
>>> c["for the fish"]-=5
>>> c
Counter({'and thank you': 42, 'goodbye': 1, 'for the fish': -5})

一些奖金功能

>>> c.most_common()[:2]
[('and thank you', 42), ('goodbye', 1)]

有关更多信息,请参阅PyMOTW - 集合 - 容器数据类型和Python文档 - 集合

Answer 3

我觉得它使用起来更优雅partial:

import functools
dd_int = functools.partial(defaultdict, int)
defaultdict(dd_int)

当然,这与lambda相同.

Answer 4

以前的答案已经解决了如何制作两级或 n 级defaultdict。在某些情况下，你想要一个无限的：

def ddict():
    return defaultdict(ddict)

用法：

>>> d = ddict()
>>> d[1]['a'][True] = 0.5
>>> d[1]['b'] = 3
>>> import pprint; pprint.pprint(d)
defaultdict(<function ddict at 0x7fcac68bf048>,
            {1: defaultdict(<function ddict at 0x7fcac68bf048>,
                            {'a': defaultdict(<function ddict at 0x7fcac68bf048>,
                                              {True: 0.5}),
                             'b': 3})})

Answer 5

作为参考，可以通过以下方式实现通用的嵌套defaultdict工厂方法：

from collections import defaultdict
from functools import partial
from itertools import repeat


def nested_defaultdict(default_factory, depth=1):
    result = partial(defaultdict, default_factory)
    for _ in repeat(None, depth - 1):
        result = partial(defaultdict, result)
    return result()

深度定义了default_factory使用中定义的类型之前嵌套字典的数量。例如：

my_dict = nested_defaultdict(list, 3)
my_dict['a']['b']['c'].append('e')

Answer 6

其他人已正确回答了您如何使以下工作的问题:

for x in stuff:
    d[x.a][x.b] += x.c_int

另一种方法是使用元组作为键:

d = defaultdict(int)
for x in stuff:
    d[x.a,x.b] += x.c_int
    # ^^^^^^^ tuple key

这种方法的好处是它很简单,可以很容易地扩展.如果您需要三级深度映射,只需使用三项元组作为密钥.