填充字典词典时出现KeyError

时间:2020-11-12 06:43:01

标签: python python-3.x dictionary defaultdict

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

cable_footage = {"AER": {144: len1, 48: len2, 24: len3}, "UG": {144: len1, 48: len2, 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。我该如何克服?

干杯!

1 个答案:

答案 0 :(得分:5)

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

尝试一下:

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

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

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