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

XlsxWriter 是一个用 Python 编写的用于创建 Excel XLSX 文件的模块. 该模块被广泛应用于以编程方式生成 Excel 工作表，包括格式设置、图表和公式等。

本文将介绍如何使用 XlsxWriter 模块来创建 Excel 工作表，并绘制简单的股票图表。

安装 XlsxWriter 模块

使用 pip 命令来安装 XlsxWriter 模块：

pip install XlsxWriter

创建 Excel 工作表

使用 XlsxWriter 模块创建 Excel 工作表很简单。下面的代码创建了一个名为 StockChart.xlsx 的 Excel 工作表，并将数据写入单元格。

import xlsxwriter
  
# Create a new workbook and add a worksheet
workbook = xlsxwriter.Workbook('StockChart.xlsx')
worksheet = workbook.add_worksheet()
  
# Add some data to the worksheet
data = [10, 40, 50, 20, 10, 50]
  
# Write the data to the worksheet
for i in range(len(data)):
    worksheet.write(i, 0, data[i])
  
# Close the workbook
workbook.close()

运行该代码，在目录下可以看到生成了一个名为 StockChart.xlsx 的 Excel 工作表。

绘制股票图表

现在，我们已经创建了一个简单的 Excel 工作表。接下来的任务是如何在工作表中创建一个股票图表。

下面的代码演示了如何在 Excel 工作表上创建一个简单的股票图表。首先，我们需要引入 chart 模块和 stock_chart 方法。然后，我们需要指定要绘制的数据范围和图表类型。

import xlsxwriter
from xlsxwriter import chart
  
workbook = xlsxwriter.Workbook('StockChart.xlsx')
worksheet = workbook.add_worksheet()
  
# Add some data to the worksheet
data = [10, 40, 50, 20, 10, 50]
for i in range(len(data)):
    worksheet.write(i, 0, data[i])
  
# Create a new chart object and specify the chart type
chart1 = workbook.add_chart({'type': 'stock'})
  
# Add series to the chart
# The first series is a simple, static line.
chart1.add_series({
    'values': '=Sheet1!A1:A' + str(len(data)),
    'name': 'Stock Prices'
    })
  
# Add chart title and axis labels
chart1.set_title({'name': 'Stock Prices'})
chart1.set_x_axis({'name': 'Date'})
chart1.set_y_axis({'name': 'Price'})
  
# Insert the chart into the worksheet
worksheet.insert_chart('C1', chart1)
  
# Close the workbook
workbook.close()

运行该代码后，打开 Excel 工作表 StockChart.xlsx，可以看到绘制了一个基本的股票图表。

总结

本文介绍了如何使用 XlsxWriter 模块来创建 Excel 工作表并绘制股票图表。有了 XlsxWriter 和 Python，你可以轻松地将数据从计算机程序中传输到 Excel 工作表中，进行进一步的数据可视化和数据分析。