python - 如果键匹配条件，则将字典的值合并为一个

Question

我有一本名为 tempDict 的字典，其填充如下:

tempDict = {'a': 100, 'b': 200, 'c': 120, 'b_ext1': 4, 'd': 1021, 'a_ext1': 21, 'f_ext1': 12}

在我的设置中，我需要循环遍历键，如果对于任何具有“_ext1”后缀的键，我想重写或创建一个新字典，保留未更改的键(最好没有“ext1”)，但使用值(value)观合并了。

即:

newDict = {'a': 121, 'b': 204, 'c': 120, 'd': 1021, 'f_ext1':12}

注意，字典中的最后一个条目应该保持不变，因为没有 'f' unsuffixed 与 '_ext1'

值本身不是整数，但操作类似。

有人有什么想法吗？

Answer 1

newDict = {}
for k in tempDict:
    if k.endswith("_ext1") and k[:-5] in tempDict:
       newDict[k[:-5]] = newDict.get(k[:-5],0)+tempDict[k]
    else:
       newDict[k] = newDict.get(k,0)+tempDict[k]