matplotlib学习笔记（5）---堆叠图

matplotlib学习笔记（5）—堆叠图

import numpy as np
import matplotlib.pyplot as plt
countries = ['Norway', 'Germany', 'USA', 'Austria', 'Netherlands'] 
bronzes = np.array([8, 3, 4, 4, 3]) 
silvers = np.array([7, 6, 7, 7, 4]) 
golds = np.array([12, 9, 8, 6, 6]) 
# 得到[0,1,2,3,4]索引
ind = [x for x, _ in enumerate(countries)] 
#绘制堆叠图
plt.bar(ind, golds, width=0.5, label='golds', color='gold', bottom=silvers+bronzes) 
plt.bar(ind, silvers, width=0.5, label='silvers', color='silver', bottom=bronzes) 
plt.bar(ind, bronzes, width=0.5, label='bronzes', color='#CD853F') 
#设置坐标轴
plt.xticks(ind, countries) 
plt.ylabel("Medals") 
plt.xlabel("Countries") 
plt.legend(loc="upper right") 
plt.title("2022 Beijing Winter Olympics")
plt.show()