如何在matplotlib中创建堆叠条形图

import matplotlib.pyplot as plt
import numpy as np

categories = ['A', 'B', 'C', 'D', 'E']  # 类别
data1 = [10, 15, 7, 12, 9]  # 数据1
data2 = [8, 12, 10, 5, 6]  # 数据2
data3 = [5, 9, 11, 6, 8]  # 数据3

fig, ax = plt.subplots()

# 设置条形图的位置和宽度
bar_width = 0.35
bar_positions = np.arange(len(categories))

# 绘制每个类别的条形图
ax.bar(bar_positions, data1, bar_width, label='Data 1')
ax.bar(bar_positions, data2, bar_width, bottom=data1, label='Data 2')
ax.bar(bar_positions, data3, bar_width, bottom=np.array(data1) + np.array(data2), label='Data 3')

# 设置图表标题和坐标轴标签
ax.set_title('Stacked Bar Chart')
ax.set_xlabel('Categories')
ax.set_ylabel('Values')

# 设置类别标签
ax.set_xticks(bar_positions)
ax.set_xticklabels(categories)

# 添加图例
ax.legend()

# 显示图表
plt.show()