Python: defaultdict из defaultdict?

Есть ли способ defaultdict(defaultdict(int)), чтобы заставить следующий код работать?

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

dдолжен быть построен Ad-Hoc, в зависимости от x.aи x.bэлементов.

Я мог бы использовать:

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

но тогда я не смог бы использовать:

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

Да, вот так:

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().

Параметром конструктора defaultdict является функция, которая будет вызываться для создания новых элементов. Так что давайте использовать лямбду!

>>> 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 - коллекции

Я нахожу это немного более элегантным для использования partial:

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

Конечно, это то же самое, что лямбда.

Для справки, можно реализовать общий вложенный 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')

В предыдущих ответах говорилось, как сделать два уровня или 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})})

Другие правильно ответили на ваш вопрос о том, как заставить работать следующее:

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

Преимущество этого подхода в том, что он прост и может быть легко расширен. Если вам нужно отображение глубиной в три уровня, просто используйте кортеж из трех предметов для ключа.