下面是一个示例代码,演示了如何按照一个字段对对象进行分组,并对另一个字段进行求和,然后生成一个CSV报告。
假设我们有一个包含"姓名"和"销售额"的对象列表,我们想要按照"姓名"字段对对象进行分组,并对每个组的销售额求和,然后生成一个CSV报告。
import csv
from collections import defaultdict
# 假设这是我们的对象列表,每个对象包含"姓名"和"销售额"字段
objects = [
{"姓名": "张三", "销售额": 100},
{"姓名": "李四", "销售额": 200},
{"姓名": "张三", "销售额": 300},
{"姓名": "王五", "销售额": 400},
{"姓名": "李四", "销售额": 500},
]
# 使用defaultdict创建一个空的字典,值的类型是列表
grouped_objects = defaultdict(list)
# 将对象按照"姓名"字段进行分组
for obj in objects:
name = obj["姓名"]
grouped_objects[name].append(obj)
# 创建一个空的CSV文件,用于存储报告
with open("report.csv", "w", newline="") as csvfile:
writer = csv.writer(csvfile)
# 写入CSV文件的表头
writer.writerow(["姓名", "销售总额"])
# 对每个组进行循环
for name, group in grouped_objects.items():
# 对组内的销售额进行求和
total_sales = sum(obj["销售额"] for obj in group)
# 将姓名和销售总额写入CSV文件
writer.writerow([name, total_sales])
上述代码将生成一个名为"report.csv"的CSV文件,其中包含每个姓名和其对应的销售总额。
注意:在运行代码之前,需要确保已安装Python的csv模块,可以使用以下命令进行安装:
pip install csv