如何使用Python中的openpyxl创建Excel中的图表？

在处理大量数据时，将数据呈现为图表是一种有效的方式。在Excel中，可以使用“插入图表”功能将数据转换为图表，但在有些情况下，需要以程序的方式自动创建图表。openpyxl是一个Python库，可以创建和编辑Excel文件。本文将介绍如何使用openpyxl创建Excel中的图表。

更多Python文章，请阅读：Python 教程

安装openpyxl

首先，需要安装openpyxl库。在命令行中使用以下命令进行安装：

pip install openpyxl

创建工作簿和工作表

使用openpyxl创建Excel文件的第一步是创建工作簿和工作表。以下示例代码创建一个名为“my_chart.xlsx”的工作簿，并在其中创建一个名为“Sheet1”的工作表：

import openpyxl

workbook = openpyxl.Workbook()
worksheet = workbook.active
worksheet.title = "Sheet1"

workbook.save(filename="my_chart.xlsx")

添加数据

接下来，需要向工作表中添加数据。以下示例代码添加了一些示例数据：

data = [("Month", "Sales"), ("January", 10000), ("February", 15000), ("March", 20000), ("April", 25000), ("May", 30000)]

for row in data:
    worksheet.append(row)

workbook.save(filename="my_chart.xlsx")

创建图表

使用openpyxl创建图表的第一步是确定图表类型。以下示例代码创建一个条形图：

from openpyxl.chart import BarChart, Reference

chart = BarChart()
chart.type = "col"
chart.title = "Sales by Month"
chart.x_axis_title = "Month"
chart.y_axis_title = "Sales"

data = Reference(worksheet, min_col=2, min_row=1, max_col=2, max_row=6)
categories = Reference(worksheet, min_col=1, min_row=2, max_row=6)

chart.add_data(data)
chart.set_categories(categories)

worksheet.add_chart(chart, "A8")

workbook.save(filename="my_chart.xlsx")

在上述示例代码中，使用Reference类引用工作表中的数据范围并添加到图表中。set_categories方法用于设置图表的类别，即x轴的值。

创建其他类型的图表

除了条形图，还可以创建其他类型的图表，如线形图、散点图和饼图。以下示例代码创建一个饼图：

from openpyxl.chart import PieChart, Reference

chart = PieChart()
chart.title = "Sales by Month"

data = Reference(worksheet, min_col=2, min_row=1, max_col=2, max_row=6)
categories = Reference(worksheet, min_col=1, min_row=2, max_row=6)

chart.add_data(data)
chart.set_categories(categories)

worksheet.add_chart(chart, "A8")

workbook.save(filename="my_chart.xlsx")