Python|使用 XlsxWriter 模块使用数据工具在 Excel 表中绘制图表 |套装 – 2

本文介绍了如何使用 Python 的 XlsxWriter 模块来在 Excel 表格中绘制图表。XlsxWriter 是一个功能强大的库，可以帮助开发者以编程方式生成 Excel 文件，并添加图表和其他自定义元素。

安装和配置环境

首先，我们需要安装 XlsxWriter 模块。可以使用以下命令在命令行中安装：

pip install XlsxWriter

安装完成后，我们可以导入 XlsxWriter 模块来使用它的功能：

import xlsxwriter

创建并编辑 Excel 表格

首先，我们需要创建一个新的 Excel 文件，并添加一些数据。

# 创建一个新的 Excel 文件
workbook = xlsxwriter.Workbook('data.xlsx')
worksheet = workbook.add_worksheet()

# 向表格中添加数据
data = [['Name', 'Salary'],
        ['John Smith', 5000],
        ['Jane Doe', 6000],
        ['Mike Johnson', 4000]]

for row_num, row_data in enumerate(data):
    for col_num, col_data in enumerate(row_data):
        worksheet.write(row_num, col_num, col_data)

workbook.close()

此代码将创建一个名为 'data.xlsx' 的新 Excel 文件，并添加一个名为 'Sheet1' 的工作表。然后，我们将数据写入工作表中。

绘制图表

可以使用 XlsxWriter 模块的图表功能来在 Excel 表格中绘制图表。以下代码演示了如何创建一个柱状图：

# 创建一个柱状图
chart = workbook.add_chart({'type': 'column'})

# 设置图表的数据范围
chart.add_series({
    'values': '=Sheet1!$B$2:$B$4',
    'categories': '=Sheet1!$A$2:$A$4',
})

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

workbook.close()

此代码将在 Excel 表格中创建一个柱状图，该图表使用工作表 'Sheet1' 中的数据。图表的数据范围是 B2:B4 单元格的值作为数据，A2:A4 单元格的值作为类别。图表将插入到工作表的 D2 单元格中。

自定义图表样式

可以根据需要自定义图表的样式。以下代码演示了如何设置图表的标题和数据系列的颜色：

# 自定义图表样式
chart.set_title({'name': 'Salary Data'})
chart.set_style(11)

# 设置数据系列的颜色
chart.set_series({
    'fill': {'color': '#FF0000'}
})

workbook.close()

此代码将设置图表的标题为 'Salary Data'，并将图表的样式设置为 11（预定义样式）。此外，数据系列的颜色被设置为红色。

总结

本文介绍了如何使用 XlsxWriter 模块在 Excel 表格中绘制图表。我们首先学习了如何创建和编辑 Excel 表格，然后演示了如何使用 XlsxWriter 的图表功能创建柱状图。最后，我们了解了如何自定义图表的样式。

XlsxWriter 是一个功能强大且灵活的工具，使得开发者可以通过 Python 代码自动化生成复杂的 Excel 文件，并添加图表、图像和其他自定义元素。它是一个非常有用的工具，特别是在需要自动化 Excel 数据处理和报告生成的项目中。