Python: defaultdict of defaultdict?

Question

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

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

d tiene que ser construido ad hoc, dependiendo de x.a y x.b elementos.

podría utilizar:

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

pero entonces yo no sería capaz de usar:

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

Answer 1

Sí así:

defaultdict(lambda: defaultdict(int)) 

El argumento de un defaultdict (en este caso es lambda: defaultdict(int)) se invocará 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 valor predeterminado, es decir, d[Key_doesnt_exist][Key_doesnt_exist], devolverá 0, que es el valor de retorno del argumento del último incumplimiento, por ejemplo, int().

Answer 2

El parámetro para el constructor de defaultdict es la función que será llamada 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 

Como Python 2.7, hay una even better solution using 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 de la prima

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

Para obtener más información, véase PyMOTW - Collections - Container data types y Python Documentation - collections

Answer 3

Me resulta un poco más elegante de use partial :

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

Por supuesto, esto es lo mismo que un lambda.

Answer 4

Otros han respondido correctamente a su pregunta de cómo conseguir lo siguiente para el trabajo:

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

Una alternativa sería utilizar tuplas para las llaves:

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 puede expandirse fácilmente. Si necesita un mapeo de tres niveles de profundidad, solo use una tupla de tres elementos para la clave.