import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
data = pd.read_excel('data_1.xlsx')
salary1 = np.array(data['MTT1'])
salary2 = np.array(data['MTT2'])
salary3 = np.array(data['MTC'])
salary4 = np.array(data['BBSQ'])
salary5 = np.array(data['Others'])
index = np.arange(len(data['横轴']))
name = data['横轴']
print(data)
def add_text(x, y, data):
for x0, y0, data0 in zip(x, y, data):
axs.text(x0, y0, round(data0, 2), rotation=0)
width = 0.6
fig, axs = plt.subplots(1, 1, figsize=(6, 6))
sum = salary1 + salary2 + salary3 + salary4+salary5
percentage1 = salary1 / sum
percentage2 = salary2 / sum
percentage3 = salary3 / sum
percentage4 = salary4 / sum
percentage5 = salary5 / sum
axs.bar(index, percentage1, width=width, label=salary1)
axs.bar(index, percentage2, width=width, bottom=percentage1, label=salary2)
axs.bar(index, percentage3, width=width,bottom=percentage1 + percentage2, label=salary3)
axs.bar(index, percentage4, width=width,bottom=percentage1 + percentage2+percentage3, label=salary4)
axs.bar(index, percentage5, width=width,bottom=percentage1 + percentage2+percentage3+percentage4, label=salary5)
axs.set_ylim(0, 1)
axs.set_xticks(index)
axs.set_xticklabels(name, rotation=50)
y1 = percentage1/2
y2 = percentage1 + percentage2/2
y3 = percentage1 + percentage2+percentage3/3
y4 = percentage1 + percentage2+percentage3+percentage4/3
y5=percentage1 + percentage2+percentage3+percentage4+percentage5/2
add_text(index-width/4, y1, percentage1)
add_text(index-width/4, y2, percentage2)
add_text(index-width/4, y3, percentage3)
add_text(index-width/4, y4, percentage4)
# add_text(index-width/2, y5, percentage5)
plt.savefig('9.png', dpi=600)
plt.show()