如何根据堆叠段高度控制堆叠条形图的排序？

import matplotlib.pyplot as plt
import numpy as np

# 示例数据
categories = ['A', 'B', 'C', 'D']
sub_categories = ['X', 'Y', 'Z']
data = {
    'A': {'X': 30, 'Y': 20, 'Z': 10},
    'B': {'X': 15, 'Y': 25, 'Z': 30},
    'C': {'X': 25, 'Y': 30, 'Z': 15},
    'D': {'X': 10, 'Y': 15, 'Z': 25}
}

# 计算每个类别的总和
totals = {cat: sum(data[cat].values()) for cat in categories}

# 按照总和排序
sorted_categories = sorted(categories, key=lambda x: totals[x], reverse=True)

# 生成堆叠条形图
fig, ax = plt.subplots()
bottom = np.zeros(len(sorted_categories))

for sub_cat in sub_categories:
    values = [data[cat][sub_cat] for cat in sorted_categories]
    ax.bar(sorted_categories, values, label=sub_cat, bottom=bottom)
    bottom += np.array(values)

ax.set_xlabel('Categories')
ax.set_ylabel('Values')
ax.set_title('Stacked Bar Chart Sorted by Total Height')
ax.legend()

plt.show()