使用Python计算各部门薪资总和的实现方案咨询

原代码问题分析 #

你编写的代码无法实现需求有两个核心原因：

1. 你设想的嵌套字典结构本身不符合Python语法规则：外层字典的元素必须是键值对，不能直接存储无对应键的子字典。
2. 字典的update()方法是用新字典的键值对覆盖旧字典同键的内容，你每次循环都在覆盖newDict中dept_id和salary两个键的值，最终只会保留最后一条数据的结果，无法实现累加。

实现部门薪资求和的方案 #

不需要复杂的嵌套结构，直接用普通字典就能完成统计，逻辑清晰且运行高效：

departments = [[10,1000],[20,2000],[30,1000],[30,4000],[10,3000]]
salary_sum = {}
# 遍历所有部门薪资数据
for dept_id, salary in departments:
    if dept_id in salary_sum:
        # 部门已存在则累加薪资
        salary_sum[dept_id] += salary
    else:
        # 部门不存在则初始化总薪资
        salary_sum[dept_id] = salary
# 转换为要求的列表格式并按部门ID排序
result = sorted([[dept, total] for dept, total in salary_sum.items()])
print(result)

运行后直接输出你需要的结果：

[[10, 4000], [20, 2000], [30, 5000]]

如果想要更简洁的写法，可以用collections模块的defaultdict，无需手动判断键是否存在：

from collections import defaultdict
departments = [[10,1000],[20,2000],[30,1000],[30,4000],[10,3000]]
salary_sum = defaultdict(int)
for dept_id, salary in departments:
    salary_sum[dept_id] += salary
result = sorted([[dept, total] for dept, total in salary_sum.items()])
print(result)

内容的提问来源于stack exchange，提问作者Dikshit Karki