forma abreviada para crear la clave del diccionario si no existe

Question

Tengo un diccionario de animales de zoológico. Quiero ponerlo en el diccionario en un diccionario anidado pero obtengo un KeyError porque esa especie en particular no se ha agregado al diccionario.

def add_to_world(self, species, name, zone = 'retreat'): 
    self.object_attr[species][name] = {'zone' : zone} 

¿Hay un acceso directo a la comprobación de si la especie se encuentra en el diccionario y crearlo si no lo es, o tengo que hacerlo de la manera larga y comprobar manualmente si es que las especies se ha añadido?

Answer 1

def add_to_world(self, species, name, zone = 'retreat'): 
    self.object_attr.setdefault(species, {})[name] = {'zone' : zone} 

Answer 2

La autovigilancia de los valores del diccionario se puede realizar por collections.defaultdict.

Answer 3

Aquí hay un ejemplo de uso de default con un diccionario como valor.

>>> from collections import defaultdict 
>>> d = defaultdict(dict) 
>>> d["species"]["name"] = {"zone": "1"} 
>>> d 
defaultdict(<type 'dict'>, {'species': {'name': {'zone': '1'}}}) 
>>> 

si quieres anidación más tendrá que hacer una función para volver defaultdict (dict).

def nested_defaultdict(): 
    return defaultdict(dict) 

# Then you can use a dictionary nested to 3 levels 
d2 = defaultdict(nested_defaultdict) 
d2["species"]["name"]["zone"] = 1