以下是一个报告生成器的代码示例,将表格结果打印到Excel的不同工作表中:
import pandas as pd
from openpyxl import Workbook
def generate_report(data):
# 创建一个新的Excel工作簿
workbook = Workbook()
for sheet_name, sheet_data in data.items():
# 在工作簿中创建一个新的工作表
sheet = workbook.create_sheet(title=sheet_name)
# 将数据添加到工作表中
for row_index, row_data in enumerate(sheet_data, start=1):
for col_index, cell_data in enumerate(row_data, start=1):
sheet.cell(row=row_index, column=col_index, value=cell_data)
# 保存工作簿为Excel文件
workbook.save("report.xlsx")
# 示例数据
data = {
"Sheet1": [
["姓名", "年龄", "性别"],
["张三", 25, "男"],
["李四", 30, "女"],
["王五", 28, "男"]
],
"Sheet2": [
["产品", "销量", "利润"],
["A", 100, 500],
["B", 200, 1000],
["C", 150, 750]
]
}
# 生成报告
generate_report(data)
以上代码使用pandas库和openpyxl库来实现报告生成器。它首先创建一个新的Excel工作簿,然后遍历数据字典中的每个表格数据。对于每个表格数据,它会在工作簿中创建一个新的工作表,并将数据添加到工作表中。最后,它将工作簿保存为Excel文件。这个示例代码生成的Excel文件中有两个工作表,分别是"Sheet1"和"Sheet2",每个工作表中包含相应的表格数据。