Python XlsxWriter - 使用 Pandas

Pandas 是一个流行的 Python 库，用于数据操作和分析。我们可以使用 XlsWriter 将 Pandas dataframes 写入 Excel 工作表。

要了解本节中描述的功能，我们需要在已安装 XlsxWriter 的相同环境中安装 Pandas 库。

pip3 install pandas

在 Pandas 中使用 XlsxWriter

让我们从一个简单的例子开始。首先，根据整数列表中的数据创建一个 Pandas 数据框。然后使用 XlsxWriter 作为引擎来创建 Pandas XlsxWriter。借助此引擎对象，我们可以将数据框对象写入 Excel 工作表。

示例

import pandas as pd

df = pd.DataFrame({'Data': [10, 20, 30, 20, 15, 30, 45]})
writer = pd.ExcelWriter('hello.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1')

writer.save()

输出

这样创建的工作表显示如下 −

向 Pandas Dataframe 添加图表

正如我们通过调用它的add_worksheet() 方法获得Workbook 类的对象，然后获得Worksheet 对象一样，也可以使用writer 对象来获取这些对象。一旦我们得到它们，就可以使用XlsxWriter方法添加图表、数据表等。

在此示例中，我们设置了一个 Pandas 数据框并获取其维度（或形状）。

import pandas as pd
df = pd.DataFrame({'Data': [105, 60, 35, 90, 15, 30, 75]})
writer = pd.ExcelWriter('hello.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1')
(max_row, max_col) = df.shape

工作簿和工作表对象是从编写器创建的。

workbook = writer.book
worksheet = writer.sheets['Sheet1']

剩下的事情很简单。正如我们之前所做的那样添加图表对象。

chart = workbook.add_chart({'type': 'column'})
chart.add_series({'values': ['Sheet1', 1, 1, max_row, 1]})
worksheet.insert_chart(1, 3, chart)
writer.save()

示例

以下代码使用Pandas dataframe编写Excel工作簿，柱状图由XlsxWriter编写。

import pandas as pd

df = pd.DataFrame({'Data': [105, 60, 35, 90, 15, 30, 75]})
writer = pd.ExcelWriter('hello.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1')
(max_row, max_col) = df.shape

workbook = writer.book
worksheet = writer.sheets['Sheet1']
chart = workbook.add_chart({'type': 'column'})
chart.add_series({'values': ['Sheet1', 1, 1, max_row, 1]})
worksheet.insert_chart(1, 3, chart)

writer.save()

输出

柱状图连同数据如下图所示 −

将Dataframe写入Excel表格

同样，dataframe可以写入Excel表格对象。此处的数据框源自 Python 字典，其中键是数据框列标题。每个键都有一个值列表，该值又成为每列的值。

import pandas as pd

df = pd.DataFrame({
   'Name': ['Namrata','Ravi','Kiran','Karishma'],
   'Percent': [73.33, 70, 75, 65.5],
   'RollNo': [1, 2,3,4]})
   
df = df[['RollNo', 'Name', 'Percent']]
(max_row, max_col) = df.shape

使用xlsxwriter引擎将dataframe写入工作表（sheet1）

writer = pd.ExcelWriter('hello.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', startrow=1, header=False, index=False)

以下几行给出了 Workbook 和 Worksheet 对象。

workbook = writer.book
worksheet = writer.sheets['Sheet1']

借助 add_table() 方法将工作表中的数据转换为表格。

column_settings = [{'header': column} for column in df.columns]

worksheet.add_table(0, 0, max_row, max_col - 1, {'columns': column_settings})

writer.save()

示例

下面是将 pandas dataframe 写入 Excel 表格的完整代码。

import pandas as pd
df = pd.DataFrame({
   'Name': ['Namrata','Ravi','Kiran','Karishma'],
   'Percent': [73.33, 70, 75, 65.5],
   'RollNo': [1, 2,3,4]
})
   
df = df[['RollNo', 'Name', 'Percent']]
(max_row, max_col) = df.shape

writer = pd.ExcelWriter('hello.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', startrow=1, header=False, index=False)

workbook = writer.book
worksheet = writer.sheets['Sheet1']

column_settings = [{'header': column} for column in df.columns]

worksheet.add_table(0, 0, max_row, max_col - 1, {'columns': column_settings})

writer.save()