使用Python创建并保存Excel文件到字节串

pip install spire.xls.free

from spire.xls import Workbook

# 创建一个新的工作簿
workbook = Workbook()

# 移除默认创建的空工作表
workbook.Worksheets.Clear()

# 添加一个新的工作表，命名为“销售数据”
sheet = workbook.Worksheets.Add("销售数据")

# 写入表头
sheet.Range["A1"].Value = "产品名称"
sheet.Range["B1"].Value = "销售额"
sheet.Range["C1"].Value = "销售日期"

# 示例数据
data = [
    ("笔记本电脑", 1200.50, "2023-10-01"),
    ("智能手机", 800.75, "2023-10-05"),
    ("平板电脑", 500.00, "2023-10-10"),
    ("显示器", 300.20, "2023-10-15"),
    ("键盘", 75.99, "2023-10-20"),
]

# 写入数据
for i, row_data in enumerate(data):
    row_index = i + 2  # 从第二行开始写入数据
    sheet.Range[row_index, 1].Value = row_data[0]  # 产品名称
    sheet.Range[row_index, 2].NumberValue = row_data[1]  # 销售额（数字类型）
    sheet.Range[row_index, 3].Value = row_data[2]  # 销售日期

# 设置表头字体加粗
sheet.Range["A1:C1"].Style.Font.IsBold = True

# 自动调整列宽
sheet.AutoFitColumn(1)
sheet.AutoFitColumn(2)
sheet.AutoFitColumn(3)

import io
from spire.xls import FileFormat, Stream

# 创建 BytesIO 对象
excel_stream = io.BytesIO()

# 创建 Stream 对象
excelStream = Stream()

# 将工作簿保存到内存流中，格式为 XLSX（FileFormat.Version2016）
workbook.SaveToStream(excelStream, FileFormat.Version2013)

# 将内存流保存到文件用于测试
excelStream.Save("output/output.xlsx")

# 将内存流转换为字节串
excel_stream.write(excelStream.ToArray())

# 获取字节串
excel_bytes = excel_stream.getvalue()

# 释放资源
workbook.Dispose()
excel_stream.close()

print(f"生成的Excel字节串长度: {len(excel_bytes)} 字节")