claves de diccionario Generación sobre la marcha

Question

Trabajar con predice pitón profundamente anidados, me gustaría ser capaz de asignar valores en una estructura de datos como esto:

mydict[key][subkey][subkey2]="value" 

sin tener que comprobar que mydict [clave], etc. .de hecho están configurados para ser un dict, por ejemplo utilizando

if not key in mydict: mydict[key]={} 

La creación de subdiccionarios debe realizarse sobre la marcha. ¿Cuál es la manera más elegante de permitir algo equivalente, tal vez usando decoradores en el estándar <type 'dict'>?

Answer 1

class D(dict): 
    def __missing__(self, key): 
     self[key] = D() 
     return self[key] 

d = D() 
d['a']['b']['c'] = 3 

Answer 2

se puede utilizar una tupla como la clave para el dict y luego usted no tiene que preocuparse en absoluto subdictionaries:

mydict[(key,subkey,subkey2)] = "value" 

Alternativamente, si usted realmente necesita tener subdictionaries por alguna razón usted podría use collections.defaultdict.

Durante dos niveles Esto es sencillo:

>>> from collections import defaultdict 
>>> d = defaultdict(dict) 
>>> d['key']['subkey'] = 'value' 
>>> d['key']['subkey'] 
'value' 

Para tres es un poco más complejo:

>>> d = defaultdict(lambda: defaultdict(dict)) 
>>> d['key']['subkey']['subkey2'] = 'value' 
>>> d['key']['subkey']['subkey2'] 
'value' 

Cuatro y más niveles se dejan como ejercicio para el lector. :-)

Answer 3

Me gusta más la respuesta de Dave, pero aquí hay una alternativa.

from collections import defaultdict 
d = defaultdict(lambda : defaultdict(int)) 
>>> d['a']['b'] += 1 
>>> d 
defaultdict(<function <lambda> at 0x652f0>, {'a': defaultdict(<type 'int'>, {'b': 1})}) 
>>> d['a']['b'] 
1 

http://tumble.philadams.net/post/85269428/python-nested-defaultdicts

Es, definitivamente, no es bonito tener que utilizar a lambdas implementa las colecciones en default interiores, pero al parecer es necesario.