按条件将Excel文件拆分到不同的工作表
我们可以使用Python的pandas库来生成这个包含多个工作表的Excel文件。以下是一个完整的代码示例,它将创建一个包含四个工作表的Excel文件:一个总成绩表(Sheet1)和三个班级表(Sheet2, Sheet3, Sheet4)。
首先,请确保您已经安装了pandas和openpyxl库。如果没有安装,可以使用以下命令进行安装:
pip install pandas openpyxl
接下来是完整的Python代码:
import pandas as pd
# 示例数据
data = {
'姓名': ['张三', '李四', '王五', '赵六', '刘七', '陈八', '钱九', '孙十', '周十一', '吴十二'],
'班级': ['三年一班', '三年一班', '三年二班', '三年二班', '三年三班', '三年三班', '三年一班', '三年二班', '三年三班',
'三年一班'],
'语文': [85, 92, 76, 89, 94, 83, 88, 79, 90, 82],
'数学': [90, 88, 81, 93, 95, 86, 92, 84, 91, 87],
'英语': [88, 91, 79, 87, 92, 85, 89, 80, 88, 86]
}
# 创建DataFrame
df = pd.DataFrame(data)
# 创建Excel writer对象
with pd.ExcelWriter('三年级成绩单.xlsx', engine='openpyxl') as writer:
# 写入总成绩表
df.to_excel(writer, sheet_name='总成绩', index=False)
# 按班级分组并写入不同的工作表
for class_name in df['班级'].unique():
class_df = df[df['班级'] == class_name]
class_df.to_excel(writer, sheet_name=class_name, index=False)
print("Excel文件已生成!")
运行结果:
代码解释
- 导入库:导入pandas库。
- 示例数据:定义一个字典
data
,包含所有学生的成绩信息。 - 创建DataFrame:使用
pd.DataFrame
将数据转换为DataFrame对象。 - 创建Excel writer对象:使用
pd.ExcelWriter
创建一个Excel writer对象,指定文件名和引擎。 - 写入总成绩表:将整个DataFrame写入名为“总成绩”的工作表。
- 按班级分组并写入不同的工作表:遍历每个班级,提取该班级的学生数据,并将其写入相应的工作表。
- 保存文件:关闭writer对象,完成文件保存。
运行上述代码后,将会在当前目录下生成一个名为“三年级成绩单.xlsx”的Excel文件,其中包含四个工作表:总成绩、三年一班、三年二班和三年三班。每个工作表中的数据按照要求进行了分类。
欢迎点赞、关注、收藏、转发!!!