将数据框保存为 Excel 文件

在数据分析过程中，我们通常需要将清洗好的数据保存成 Excel 文件，以便后续进行数据可视化或数据建模等操作。Python 中的 Pandas 库提供了非常方便的方式来实现这个功能。

1. 使用 Pandas 将数据框保存为 Excel 文件

Pandas 使用 to_excel() 函数来将数据框保存为 Excel 文件。以下是使用 Pandas 将数据框保存为 Excel 文件的示例代码：

import pandas as pd

# 创建一个数据框
df = pd.DataFrame({'Name': ['John', 'Grace', 'Alex'], 'Age': [25, 30, 35], 'Salary': [5000, 6000, 7000]})

# 将数据框保存为 Excel 文件
df.to_excel('example.xlsx', index=False)

以上代码创建了一个数据框，将其保存为名为 example.xlsx 的 Excel 文件。

该函数的参数说明如下：

• excel_writer：Excel 文件名或者是 ExcelWriter 对象。
• sheet_name：Excel 文件的工作表名。
• index：是否保存行索引。默认是 True。
• header：是否保存列名。默认是 True。

2. 使用 openpyxl 将数据框保存为 Excel 文件

如果要对 Excel 文件进行更复杂的操作，可以使用 openpyxl 库。openpyxl 库不仅可以读取和写入 Excel 文件，还可以修改 Excel 文件的内容。

以下是使用 openpyxl 将数据框保存为 Excel 文件的示例代码：

import pandas as pd
from openpyxl import Workbook

# 创建一个数据框
df = pd.DataFrame({'Name': ['John', 'Grace', 'Alex'], 'Age': [25, 30, 35], 'Salary': [5000, 6000, 7000]})

# 新建一个 Excel 文件
wb = Workbook()

# 将数据框保存为 Excel 文件
ws = wb.active
ws.title = 'example'
for r in dataframe_to_rows(df, index=False, header=True):
    ws.append(r)

# 保存 Excel 文件
wb.save('example.xlsx')

以上代码创建了一个数据框，使用 openpyxl 库新建了一个名为 example.xlsx 的 Excel 文件，并将数据框保存到该 Excel 文件。

其中，dataframe_to_rows() 函数将数据框转换为行的列表，方便写入工作表中。

3. 总结

以上介绍了两种常用的方法来将数据框保存为 Excel 文件。Pandas 库提供了非常方便的方式来实现这个功能，而 openpyxl 库则可以对 Excel 文件进行更复杂的操作。使用时，可以根据需要进行选择。