Python pandas写入Excel|极客教程

Python pandas写入Excel

在数据分析和处理中，常常需要将处理好的数据保存到Excel文件中以进行进一步的分析或者共享。Python中的pandas库提供了很方便的方法来将数据写入Excel文件中。本文将详细介绍如何使用pandas库来写入Excel文件。

安装pandas库

如果还没有安装pandas库，可以通过pip来进行安装：

pip install pandas

创建样本数据

为了演示如何写入Excel文件，我们首先创建一些样本数据。我们使用pandas的DataFrame来创建一个包含学生成绩的数据集。

import pandas as pd

data = {
    '姓名': ['小明', '小红', '小刚', '小美'],
    '语文': [80, 90, 85, 88],
    '数学': [95, 88, 92, 85],
    '英语': [70, 75, 80, 82]
}

df = pd.DataFrame(data)
print(df)

运行结果如下：

   姓名  语文  数学  英语
0  小明  80  95  70
1  小红  90  88  75
2  小刚  85  92  80
3  小美  88  85  82

写入Excel文件

接下来，我们使用pandas的to_excel方法将数据写入Excel文件。在这里，我们需要指定文件名和工作表名。

output_file = '学生成绩表.xlsx'
sheet_name = '学生成绩'

df.to_excel(output_file, sheet_name=sheet_name, index=False)

上面的代码将DataFrame df 写入了一个名为“学生成绩表.xlsx”的Excel文件中，并在工作表中命名为“学生成绩”。参数index=False表示不将索引写入Excel文件中。

查看写入结果

我们可以用Excel软件打开生成的Excel文件来查看写入的结果。在Excel中，打开“学生成绩表.xlsx”文件，可以看到如下内容：

姓名	语文	数学	英语
小明	80	95	70
小红	90	88	75
小刚	85	92	80
小美	88	85	82

指定写入位置

有时候我们可能希望将数据写入到某个特定的位置，而不是从Excel的第一个单元格开始。这时可以使用ExcelWriter对象来实现。

from openpyxl import load_workbook

output_file = '学生成绩表.xlsx'
sheet_name = '学生成绩'

with pd.ExcelWriter(output_file, engine='openpyxl') as writer:
    writer.book = load_workbook(output_file)
    writer.sheets = dict((ws.title, ws) for ws in writer.book.worksheets)
    df.to_excel(writer, sheet_name=sheet_name, startrow=2, startcol=1, index=False)