填充字典词典时出现KeyError

问题描述

我基本上需要以下格式的词典字典:

cable_footage = {"AER": {144: len1,48: len2,24: len3},"UG": {144: len1,24: len3}}

这是我到目前为止的代码

cable_footage = defaultdict(dict)
    for placement in ("AER","UG"):
        for size in (144,48,24):
            cable_footage[placement][size] = 0.0
    for cable in adss_layer.features:
        if cable.data["Placement"] == "AER":
            if cable.data["Size"] == 144:
                cable_footage["AER"][144] += cable.data["Length"]
            if cable.data["Size"] == 48:
                cable_footage["AER"][48] += cable.data["Length"]
        if cable.data["Placement"] == "UG":
            if cable.data["Size"] == 144:
                cable_footage["UG"][144] += cable.data["Length"]
            if cable.data["Size"] == 48:
                cable_footage["UG"][48] += cable.data["Length"]

但是,由于上面的代码很长且不具有Python风格,因此我希望能够以这种方式进行简化/通用化:

for cable in adss_layer.features:
    cable_footage[cable.data["Placement"]][cable.data["Size"]] += cable.data["Length"]

但是会引发KeyError。我该如何克服?

干杯!

解决方法

为了动态填充cable_footage,应将其定义为嵌套的defaultdictdefaultdict中的defaultdict)。

尝试一下:

cable_footage = defaultdict(lambda: defaultdict(float)))

然后,在嵌套循环内部填充它:

cable_footage[cable.data["Placement"]][cable.data["Size"]] += cable.data["Length"]