Python | 使用openpyxl模块在excel表格中绘制图表

简介

本文介绍了如何使用Python中的openpyxl模块在Excel表格中绘制图表。openpyxl是一个用于读取和写入Excel文件的库，而且它还提供了用于创建和编辑图表的功能。

安装openpyxl模块

要使用openpyxl模块，首先需要安装它。可以使用以下命令在命令提示符或终端中安装openpyxl模块：

pip install openpyxl

创建Excel表格

在绘制图表之前，首先需要创建一个Excel表格。可以使用openpyxl模块中的Workbook()函数来创建一个新的Excel工作簿。以下是创建一个新工作簿的示例代码：

from openpyxl import Workbook

workbook = Workbook()
sheet = workbook.active

# 在工作簿中添加数据
sheet['A1'] = 1
sheet['A2'] = 2
sheet['A3'] = 3
sheet['B1'] = 4
sheet['B2'] = 5
sheet['B3'] = 6

# 保存工作簿
workbook.save("data.xlsx")

该代码将创建一个新的Excel工作簿，并在工作簿中添加一些数据。数据将保存在名为"data.xlsx"的文件中。

绘制图表

在创建了Excel表格后，可以使用openpyxl模块中的Chart类来创建图表。以下是在Excel表格中绘制一个简单的柱形图的示例代码：

from openpyxl import Workbook
from openpyxl.chart import BarChart

# 创建一个新的工作簿
workbook = Workbook()
sheet = workbook.active

# 在工作簿中添加数据
sheet['A1'] = 1
sheet['A2'] = 2
sheet['A3'] = 3
sheet['B1'] = 4
sheet['B2'] = 5
sheet['B3'] = 6

# 创建一个柱形图
chart = BarChart()
chart.title = "Sample Chart"
chart.x_axis.title = "X-axis"
chart.y_axis.title = "Y-axis"

# 设置图表的数据范围
data = Reference(sheet, min_col=1, min_row=1, max_col=2, max_row=3)
categories = Reference(sheet, min_col=1, min_row=2, max_row=3)

# 将数据和类别添加到图表中
chart.add_data(data, titles_from_data=True)
chart.set_categories(categories)

# 将图表添加到工作簿中
sheet.add_chart(chart, "D1")

# 保存工作簿
workbook.save("data_with_chart.xlsx")

该代码将创建一个柱形图，并将其添加到Excel表格中。柱形图将使用数据范围A1:B3中的数据，并使用A2:A3中的类别。工作簿将保存在名为"data_with_chart.xlsx"的文件中。

总结

通过使用openpyxl模块，可以轻松地在Excel表格中绘制各种类型的图表。只需创建一个Excel工作簿，将数据添加到工作簿中，然后使用Chart类创建所需的图表并将其添加到工作簿中。