按lstA元素匹配字典键分组求和，代码返回全值总和如何解决？

嘿，我来帮你搞定这个分组求和的问题！先看看你可能踩的坑，再给你正确的写法~

先分析你的错误原因 #

你提到当前代码返回了字典所有值的总和，大概率是两个问题：

• 求和的变量没有在每次处理新分组时重置，导致之前的累加值被带到下一个分组里
• 没有判断字典的键是否包含当前lstA里的元素，直接把所有值都加了一遍

举个例子，错误的代码可能是这样的：

lstA = ["apple", "banana"]
my_dict = {"apple_1": 10, "apple_2": 20, "banana_1": 15, "orange_1": 5}
total = 0
result = {}

for item in lstA:
    for key, value in my_dict.items():
        total += value
    result[item] = total

这段代码里，total在循环外初始化，每次外层循环都会把整个字典的值加一遍，最后每个分组都是所有值的总和（50），完全不符合预期。

正确的嵌套循环写法 #

只需要调整两个地方就能解决：

• 把求和变量放到外层循环内部，每次处理新分组时从0开始累加
• 增加判断条件，只对键包含当前lstA元素的值求和

代码如下：

lstA = ["apple", "banana"]
my_dict = {"apple_1": 10, "apple_2": 20, "banana_1": 15, "orange_1": 5}
result = {}

for item in lstA:
    group_sum = 0  # 每次处理新分组时重置求和变量
    for key, value in my_dict.items():
        if item in key:  # 仅累加键包含当前分组元素的值
            group_sum += value
    result[item] = group_sum

print(result)  # 输出: {'apple': 30, 'banana': 15}

更简洁的写法：字典推导式 #

如果你想让代码更紧凑，可以用字典推导式结合生成器表达式，一行搞定：

lstA = ["apple", "banana"]
my_dict = {"apple_1": 10, "apple_2": 20, "banana_1": 15, "orange_1": 5}

result = {item: sum(v for k, v in my_dict.items() if item in k) for item in lstA}
print(result)  # 输出同样是 {'apple': 30, 'banana': 15}

内容的提问来源于stack exchange，提问作者Ava Xue