Python | 使用XlsxWriter模块在Excel工作表中绘制甜甜圈图

简介

本文将介绍如何使用Python语言以及XlsxWriter模块在Excel工作表中绘制甜甜圈图。XlsxWriter是一个用于创建Excel XLSX文件的Python库，可以使用它来生成Excel电子表格，并将图表插入电子表格中。

我们将使用XlsxWriter的一个示例程序，该程序将创建一个包含甜甜圈图的Excel电子表格，其中显示了一些数据的占比情况。

准备工作

创建一个新的Python脚本，然后打开终端并在命令行中运行以下命令来安装XlsxWriter模块：

pip install XlsxWriter

绘制甜甜圈图

首先，我们需要导入XlsxWriter模块并创建一个新的Excel文件。

# 导入XlsxWriter模块
import xlsxwriter

# 创建新的Excel文件
workbook = xlsxwriter.Workbook('donut_chart.xlsx')

然后，我们需要创建一个工作表并将其命名为“Donuts”。

# 创建名为“Donuts”的工作表
worksheet = workbook.add_worksheet('Donuts')

接下来，我们需要定义数据和标签。这是我们示例数据和标签：

# 定义我们的数据和标签
data = [50, 25, 10, 15]
labels = ['苹果', '橙子', '香蕉', '草莓']

接下来，我们需要创建一个甜甜圈图，并将其插入工作表中。我们将使用XlsxWriter的add_chart()方法来实现这一点。

# 创建一个甜甜圈图
chart = workbook.add_chart({'type': 'doughnut'})

# 将数据添加到图表中
chart.add_series({
    'name': '数据',
    'categories': labels,
    'values': data,
})

# 将图表插入到工作表中
worksheet.insert_chart('B2', chart)

我们使用add_chart()方法来创建一个新的甜甜圈图，并使用数据和标签创建一个图表系列。最后，我们将图表插入到工作表的B2单元格中。

最后，我们需要关闭Excel文件以保存更改并退出程序。

# 关闭Excel文件
workbook.close()

完整代码

# 导入XlsxWriter模块
import xlsxwriter

# 创建新的Excel文件
workbook = xlsxwriter.Workbook('donut_chart.xlsx')

# 创建名为“Donuts”的工作表
worksheet = workbook.add_worksheet('Donuts')

# 定义我们的数据和标签
data = [50, 25, 10, 15]
labels = ['苹果', '橙子', '香蕉', '草莓']

# 创建一个甜甜圈图
chart = workbook.add_chart({'type': 'doughnut'})

# 将数据添加到图表中
chart.add_series({
    'name': '数据',
    'categories': labels,
    'values': data,
})

# 将图表插入到工作表中
worksheet.insert_chart('B2', chart)

# 关闭Excel文件
workbook.close()

结论

本文介绍了如何使用XlsxWriter模块在Excel工作表中绘制甜甜圈图。我们展示了如何创建一个新的Excel文件、工作表和甜甜圈图，以及如何将数据和标签添加到图表中。如果你需要在Excel中生成图表，XlsxWriter是一个非常好的Python库，应该是你的首选。