在堆叠条形图Matplotlib中注记值

import matplotlib.pyplot as plt

# 定义数据
categories = ['A', 'B', 'C', 'D']
subcategories = ['Sub1', 'Sub2', 'Sub3']
data = [[10, 20, 30],
        [15, 25, 35],
        [5, 15, 25],
        [20, 30, 40]]

# 绘制堆叠条形图
fig, ax = plt.subplots()
bars = []
bottom = [0] * len(categories)

for i, subcat_data in enumerate(data):
    bar = ax.bar(categories, subcat_data, bottom=bottom)
    bars.append(bar)
    bottom = [b + d for b, d in zip(bottom, subcat_data)]

# 注记值
for bar in bars:
    for rect in bar:
        height = rect.get_height()
        ax.text(rect.get_x() + rect.get_width() / 2, height, str(height),
                ha='center', va='bottom')

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

# 显示图表
plt.show()