Python: вирок за замовчуванням за замовчуванням?

Чи є спосіб 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()

Так ось так:

defaultdict(lambda: defaultdict(int))

Аргумент defaultdict(у даному випадку є lambda: defaultdict(int)) буде викликаний при спробі отримати доступ до ключа, який не існує. Повернене значення буде задано як нове значення цього ключа, що означає, у нашому випадку значення d[Key_doesnt_exist]буде defaultdict(int).

Якщо ви спробуєте отримати доступ до ключа з цього останнього рішення за замовчуванням, тобто d[Key_doesnt_exist][Key_doesnt_exist]він поверне 0, що є значенням повернення аргументу останнього рішення за замовчуванням, тобто int().

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

>>> 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 Documentation - колекції

Я вважаю це трохи більш елегантним у використанні 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

Приємне в цьому підході полягає в тому, що він простий і його можна легко розширити. Якщо вам потрібно картографувати три рівні глибиною, просто використовуйте кортеж з трьох елементів для ключа.