Python XlsxWriter excle日期列365天，雨量列数据自动生成柱形图

import xlsxwriter
import random

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

# 添加一个工作表
worksheet = workbook.add_worksheet()

# 添加标题行
worksheet.write('A1', 'Date')
worksheet.write('B1', 'Rainfall')

# 生成365天的日期和随机雨量数据
dates = [f'2022-01-{i:02d}' for i in range(1, 366)]
rainfall = [random.randint(0, 100) for _ in range(365)]

# 将日期和雨量数据写入工作表
for i, date in enumerate(dates):
    worksheet.write(i+1, 0, date)
for i, value in enumerate(rainfall):
    worksheet.write(i+1, 1, value)

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

# 添加数据系列
chart.add_series({
    'name': 'Rainfall',
    'categories': f'=Sheet1!$A$2:$A${len(dates)+1}',
    'values': f'=Sheet1!$B$2:$B${len(rainfall)+1}',
})

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

# 关闭Excel文件
workbook.close()