问题描述
我基本上需要以下格式的词典字典:
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
,应将其定义为嵌套的defaultdict
(defaultdict
中的defaultdict
)。
尝试一下:
cable_footage = defaultdict(lambda: defaultdict(float)))
然后,在嵌套循环内部填充它:
cable_footage[cable.data["Placement"]][cable.data["Size"]] += cable.data["Length"]