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

介绍

XlsxWriter 是一款用于创建 Excel XLSX 文件的 Python 模块，可以直接使用 Python 语言创建 Excel 文件，并在其中插入图表、图形和图像等元素。

本篇文章介绍如何使用 XlsxWriter 模块在 Excel 工作表中绘制条形图。通过本文的学习，您将能够编写 Python 代码，将数据转化为条形图并插入到 Excel 工作表中。

准备工作

在开始编写代码之前，请确保您已安装了 XlsxWriter 模块。您可以使用 pip install 命令进行安装:

pip install XlsxWriter

代码实现

下面是使用 XlsxWriter 模块在 Excel 工作表中绘制条形图的示例代码。代码中创建了一个 Excel 工作簿，并向其中插入了一张表格和一张条形图。图表使用了样式进行美化。

import xlsxwriter

# 创建一个 Excel 工作簿
workbook = xlsxwriter.Workbook('chart.xlsx')

# 新建一个工作表
worksheet = workbook.add_worksheet()

# 设置表格数据
headings = ['日期', '销售量']
data = [
    ['2021-01-01', 30],
    ['2021-01-02', 50],
    ['2021-01-03', 70],
    ['2021-01-04', 40],
    ['2021-01-05', 80],
]

# 写入表格头部
for i, heading in enumerate(headings):
    worksheet.write(0, i, heading)

# 写入表格数据
for i, row_data in enumerate(data):
    for j, cell_data in enumerate(row_data):
        worksheet.write(i+1, j, cell_data)

# 新建一个条形图
chart = workbook.add_chart({'type': 'bar'})

# 添加数据
chart.add_series({
    'name': '销售量',
    'categories': '=Sheet1!$A$2:$A$6',
    'values': '=Sheet1!$B$2:$B$6'
})

# 设置图表样式
chart.set_title({'name': '销售量'})
chart.set_x_axis({'name': '日期'})
chart.set_y_axis({'name': '销售量'})
chart.set_legend({'position': 'none'})

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

# 关闭工作簿
workbook.close()

结果演示

运行上述代码后，会生成一个名为 chart.xlsx 的 Excel 文件，其中含有一个名为 Sheet1 的工作表。工作表中包括一张表格和一张条形图，如下图所示:

总结

本篇文章介绍了如何使用 XlsxWriter 模块在 Excel 工作表中绘制条形图，并演示了结果。希望对您有所帮助。如果您对本文内容有任何疑问或建议，请随时与我们联系。