如何在matplotlib中将月份和年份数据绘制为条形图？

import matplotlib.pyplot as plt
import numpy as np

months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
years = [2018, 2019, 2020, 2021]

data = np.random.rand(len(years), len(months))

fig, ax = plt.subplots()
width = 0.2  # 设置条形宽度

# 为每个年份创建一组条形图
for i, year in enumerate(years):
    x = np.arange(len(months))  # 月份在x轴上的位置
    ax.bar(x + i*width, data[i], width, label=str(year))

# 设置坐标轴标签和标题
ax.set_ylabel('数据')
ax.set_xlabel('月份')
ax.set_title('月份和年份数据条形图')

# 设置x轴刻度标签
ax.set_xticks(x + width*(len(years)-1) / 2)
ax.set_xticklabels(months)

# 添加图例
ax.legend()

# 显示图形
plt.show()