Establecer valor dict anidado y crear claves intermedias

Question

Siento que vi una forma de hacer esto recientemente. Digamos que tengo un dict vacío y quiero establecer un valor en un dict anidado dentro de ese dict vacío, pero obviamente esa dict anidada aún no se ha creado. ¿Hay una forma de 1 línea para crear las teclas intermedias? Esto es lo que quiero hacer:

mydict = {} 
mydict['foo']['bar']['foobar'] = 25 

Si se ejecuta el código que obtendrá una excepción para KeyError 'foo'. ¿Hay una función para crear claves intermedias?

Gracias.

Answer 1

from collections import defaultdict 
recursivedict = lambda: defaultdict(recursivedict) 
mydict = recursivedict() 

Al acceder a mydict['foo'], establece mydict['foo'] a otro recursivedict. De hecho, construirá un recursivedict para mydict['foo']['bar']['foobar'], pero luego se descartará al asignarlo al 25.

Answer 2

No sé por qué te gustaría, pero:

>>> from collections import defaultdict as dd 
>>> mydict = dd(lambda: dd(lambda: {})) 
>>> mydict['foo']['bar']['foobar'] = 25 
>>> mydict 
defaultdict(<function <lambda> at 0x021B8978>, {'foo': defaultdict(<function <lambda> at 0x021B8618>, {'bar': {'foobar': 25}})}) 

Answer 3

Una opción alternativa - en función de sus usos, es el uso de tuplas como claves en lugar de los diccionarios anidados:

mydict = {} 
mydict['foo', 'bar', 'foobar'] = 25 

Este funcionará perfectamente bien a menos que desee obtener una rama del árbol en cualquier punto (en este caso no puede obtener mydict ['foo']).

Si sabía cuántas capas de anidación desea, también podría usar functools.partial en lugar de lambda.

from functools import partial 
from collections import defaultdict 

tripledict = partial(defaultdict, partial(defaultdict, dict)) 
mydict = tripledict() 
mydict['foo']['bar']['foobar'] = 25 

cual algunas personas encuentran más fácil de leer, y es más rápido para crear instancias de que la solución basada en lambda equivalentes:

python -m timeit -s "from functools import partial" -s "from collections import defaultdict" -s "tripledefaultdict = partial(defaultdict, partial(defaultdict, dict))" "tripledefaultdict()" 
1000000 loops, best of 3: 0.281 usec per loop 

python -m timeit -s "from collections import defaultdict" -s "recursivedict = lambda: defaultdict(recursivedict)" "recursivedict()" 
1000000 loops, best of 3: 0.446 usec per loop 

Aunque, como siempre, no hay ningún punto de optimización hasta que sabe que hay es un cuello de botella, así que escoja lo que es más útil y legible antes de lo que es más rápido.