python 绘制条形图

num_types = 7
count_i = [np.sum(labels[idx_train].numpy()==i) for i in range(num_types)]  # count num of each type

# 显示中文字体
matplotlib.rcParams['font.sans-serif'] = ['SimHei']
matplotlib.rcParams['axes.unicode_minus'] = False

# 绘图
rects = plt.bar(x=range(num_types), height=count_i, width=0.4, alpha=0.8, color='red', label="训练集")

# 在每个条上标注数量
for rect in rects:
    height = rect.get_height()
    plt.text(rect.get_x() + rect.get_width() / 2, height+1, str(height), ha="center", va="bottom")

plt.ylim(0,50)
plt.xlabel('节点类型')
plt.ylabel('数量')
plt.legend()
plt.show()