Python: defaultdict de defaultdict?

¿Hay alguna manera de tener un defaultdict(defaultdict(int))para que el siguiente código funcione?

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

dnecesita ser construido ad-hoc, dependiendo de x.ay x.belementos.

Podría usar:

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

pero entonces no podría usar:

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

Sí, así:

defaultdict(lambda: defaultdict(int))

Se llamará al argumento de a defaultdict(en este caso lambda: defaultdict(int)) cuando intente acceder a una clave que no existe. El valor de retorno se establecerá como el nuevo valor de esta clave, lo que significa que en nuestro caso el valor de d[Key_doesnt_exist]será defaultdict(int).

Si intenta acceder a una clave desde este último defaultdict, es decir d[Key_doesnt_exist][Key_doesnt_exist], devolverá 0, que es el valor de retorno del argumento del último defaultdict, es decir int().

El parámetro para el constructor defaultdict es la función que se llamará para construir nuevos elementos. ¡Entonces usemos una lambda!

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

Desde Python 2.7, hay una solución aún mejor con 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})

Algunas características adicionales

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

Para obtener más información, consulte PyMOTW - Colecciones - Tipos de datos de contenedor y Documentación de Python - colecciones

Me parece un poco más elegante de usar partial:

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

Por supuesto, esto es lo mismo que una lambda.

Como referencia, es posible implementar un defaultdictmétodo de fábrica anidado genérico a través de:

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

La profundidad define el número de diccionarios anidados antes de default_factoryusar el tipo definido en . Por ejemplo:

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

Las respuestas anteriores han abordado cómo hacer dos niveles o n niveles defaultdict. En algunos casos, quieres uno infinito:

def ddict():
    return defaultdict(ddict)

Uso:

>>> 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})})

Otros han respondido correctamente su pregunta sobre cómo hacer que funcione lo siguiente:

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

Una alternativa sería usar tuplas para claves:

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

Lo bueno de este enfoque es que es simple y se puede ampliar fácilmente. Si necesita un mapeo de tres niveles de profundidad, simplemente use una tupla de tres elementos para la clave.