6

我有一本动物园动物字典。我想将它放入嵌套字典中的字典中，但得到一个 KeyError，因为该特定物种尚未添加到字典中。

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

是否有快捷方式来检查该物种是否在字典中，如果不在则创建它，还是我必须做很长的路要走，然后手动检查该物种是否已添加？

4

3 回答 3

14 
def add_to_world(self, species, name, zone = 'retreat'):
    self.object_attr.setdefault(species, {})[name] = {'zone' : zone}
于 2012-06-01T01:57:22.610 回答
11

这是一个使用带有字典作为值的 defaultdict 的示例。

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

如果您想进一步嵌套，您需要创建一个函数来返回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
于 2012-06-01T01:59:40.517 回答
9

字典值的自动激活可以通过collections.defaultdict.

于 2012-06-01T01:55:32.400 回答