使用 openpyxl 将 DataFrame 写入 Excel 文件
要使用 openpyxl 将 DataFrame 写入 Excel 文件,可以按照以下步骤进行操作:
- 导入必要的库:
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rows
- 创建一个新的 Excel 工作簿:
workbook = openpyxl.Workbook()
- 选择要写入数据的工作表:
worksheet = workbook.active
- 将 DataFrame 数据转换为行的迭代器:
rows = dataframe_to_rows(df, index=False, header=True)
- 将数据逐行写入工作表:
for r_idx, row in enumerate(rows, 1):
for c_idx, value in enumerate(row, 1):
worksheet.cell(row=r_idx, column=c_idx, value=value)
- 保存 Excel 文件:
workbook.save('output.xlsx')
以下是完整的示例代码:
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rows
# 创建一个新的 Excel 工作簿
workbook = openpyxl.Workbook()
# 选择要写入数据的工作表
worksheet = workbook.active
# 假设 df 是要写入 Excel 的 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 将 DataFrame 数据转换为行的迭代器
rows = dataframe_to_rows(df, index=False, header=True)
# 将数据逐行写入工作表
for r_idx, row in enumerate(rows, 1):
for c_idx, value in enumerate(row, 1):
worksheet.cell(row=r_idx, column=c_idx, value=value)
# 保存 Excel 文件
workbook.save('output.xlsx')
这将创建一个名为'output.xlsx'的 Excel 文件,并将 DataFrame 的数据写入其中。
原文地址: https://www.cveoy.top/t/topic/l4fn 著作权归作者所有。请勿转载和采集!