xlsxwriter配合pandas|极客教程

xlsxwriter配合pandas

在数据分析和报告制作过程中，经常需要将处理好的数据导出成Excel文件以便与他人共享或者进一步的分析。而Python中的xlsxwriter和pandas库提供了一种简单而强大的方式来实现这个目标。本文将详细介绍如何使用xlsxwriter配合pandas来创建Excel文件并填充数据。

安装xlsxwriter和pandas库

首先，我们需要安装xlsxwriter和pandas库。可以通过以下命令来安装：

pip install xlsxwriter pandas

创建Excel文件并填充数据

下面我们来演示如何使用xlsxwriter和pandas来创建一个Excel文件并填充数据。

import pandas as pd

# 创建一个DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Age': [25, 30, 35, 40],
    'City': ['New York', 'Los Angeles', 'Chicago', 'Houston']
}
df = pd.DataFrame(data)

# 创建一个Excel写入器
writer = pd.ExcelWriter('output.xlsx', engine='xlsxwriter')

# 将DataFrame写入Excel
df.to_excel(writer, sheet_name='Sheet1', index=False)

# 关闭Excel写入器
writer.save()

print("Excel文件输出完成")

上面的代码首先创建了一个简单的DataFrame，并将其写入一个名为output.xlsx的Excel文件中。在代码中，我们使用了pd.ExcelWriter来创建一个Excel写入器，然后调用DataFrame的to_excel方法将数据写入Excel文件，最后关闭写入器。需要注意的是，在to_excel方法中，我们使用了index=False来避免将DataFrame的索引写入Excel文件。

填充样式和格式

除了填充数据外，xlsxwriter还提供了丰富的样式和格式设置功能，让我们可以定制Excel文件的外观。

import pandas as pd

# 创建一个DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David'],
    'Age': [25, 30, 35, 40],
    'City': ['New York', 'Los Angeles', 'Chicago', 'Houston']
}
df = pd.DataFrame(data)

# 创建一个Excel写入器
writer = pd.ExcelWriter('styled_output.xlsx', engine='xlsxwriter')

# 将DataFrame写入Excel
df.to_excel(writer, sheet_name='Sheet1', index=False)

# 获取Excel文件对象
workbook = writer.book
worksheet = writer.sheets['Sheet1']

# 创建样式
header_format = workbook.add_format({'bold': True, 'bg_color': '#F0F8FF', 'align': 'center', 'valign': 'vcenter'})
cell_format = workbook.add_format({'align': 'center', 'valign': 'vcenter'})

# 写入列标题和数据
for col_num, value in enumerate(df.columns.values):
    worksheet.write(0, col_num, value, header_format)

for row_num, row_data in df.iterrows():
    for col_num, value in enumerate(row_data):
        worksheet.write(row_num + 1, col_num, value, cell_format)

# 关闭Excel写入器
writer.save()

print("带样式和格式的Excel文件输出完成")

在上面的代码中，我们首先创建了一个Excel写入器，然后使用workbook.add_format方法创建了两种样式：header_format用于表头，cell_format用于单元格数据。之后，我们分别写入表头和数据，并指定了相应的样式。最后关闭写入器。