对堆叠条形图进行重新排序

import matplotlib.pyplot as plt
import numpy as np

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

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

# 绘制堆叠条形图
fig, ax = plt.subplots()
bottom = np.zeros(len(sorted_categories))
for i, sub_category in enumerate(sub_categories):
    values = [data[category][i] for category in sorted_categories]
    ax.bar(sorted_categories, values, bottom=bottom, label=sub_category)
    bottom += np.array(values)

ax.set_xlabel('Categories')
ax.set_ylabel('Values')
ax.set_title('Stacked Bar Chart with Reordering')
ax.legend()

plt.show()