import matplotlib.pyplot as plt
import numpy as np
import tikzplotlib

# plt.style.use("ggplot")

MCS = [ 'MI-FGSM', 'PGD', 'Adv-\nAttribute', 'Adv-\nDiffusion','TCA$^2$']
gold_medal = [ 54.82, 49.72, 50.12, 59.28,63.03]
silver_medal = [ 31.06, 24.57, 18.63, 32.61,36.08]

x = np.arange(len(MCS))
print(x)
width = 0.4
gold_x = x
silver_x = x + width
plt.bar(gold_x,gold_medal,width=width,color="royalblue", label='Face++')
plt.bar(silver_x,silver_medal,width=width,color="tomato", label='Aliyun')
# ​plt.bar(silver_x,silver_medal,width=width,color="silver")
plt.xticks(x+width/2, labels=MCS)
for i in range(len(MCS)):
    plt.text(gold_x[i],gold_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=10)
    plt.text(silver_x[i],silver_medal[i], silver_medal[i],va="bottom",ha="center",fontsize=10)
    # plt.text(bronze_x[i],bronze_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=8)  
plt.legend(loc=2,frameon=False, fontsize=10)
plt.ylabel('Mean Confidence Scores', fontsize=10)

# tikzplotlib.save("test.tex")
plt.show()

# import matplotlib.pyplot as plt
# import numpy as np

# # 这两行代码解决 plt 中文显示的问题
# # plt.rcParams['font.sans-serif'] = ['SimHei']
# # plt.rcParams['axes.unicode_minus'] = False

# # 输入统计数据
# waters = ['clean', 'MI-FGSM', 'PGD', 'Adv-\nAttribute', 'Adv-\nDiffusion','TCA^2']
# buy_number_male = [10.22, 57.82, 53.72, 50.12, 59.28,61.03]
# buy_number_female = [0.51, 31.06, 24.57, 18.63, 32.61,33.08]

# bar_width = 0.3  # 条形宽度
# index_male = np.arange(len(waters))  # 男生条形图的横坐标
# index_female = index_male + bar_width  # 女生条形图的横坐标

# # 使用两次 bar 函数画出两组条形图
# plt.bar(index_male, height=buy_number_male, width=bar_width, color='b', label='Face++')
# plt.bar(index_female, height=buy_number_female, width=bar_width, color='g', label='Aliyun')

# plt.legend()  # 显示图例
# plt.xticks(index_male + bar_width/2, waters)  # 让横坐标轴刻度显示 waters 里的饮用水， index_male + bar_width/2 为横坐标轴刻度的位置
# plt.ylabel('MCS')  # 纵坐标轴标题
# plt.title('on')  # 图形标题

# plt.show()