plot_valid_num_ratio.py

import matplotlib.pyplot as plt
import numpy as np

label_list = ['50', '100', '150', '200', '250']  # 横坐标刻度显示值

fedavg_list = np.array([184, 384, 600, 810, 994])  # 纵坐标值1
fedcs_list = np.array([116, 236, 372, 500, 616])  # 纵坐标值2
ddqn_list = np.array([183, 383, 599, 808, 995])  # 纵坐标值3
pred_list = np.array([184, 384, 600, 810, 994])  # 纵坐标值4
offline_list = np.array([184, 384, 600, 810, 994])  # 纵坐标值4

fedavg_choose_list = np.array([500, 1000, 1500, 2000, 2500])
fedcs_choose_list = np.array([158, 308, 494, 660, 818])
ddqn_choose_list = np.array([198, 413, 649, 871, 1072])
pred_choose_list = np.array([200, 414, 650, 870, 1070])
offline_choose_list = np.array([184, 384, 600, 810, 994])

err_fa_list = [8, 11, 19, 26, 31]
err_fc_list = [2, 4, 6, 8, 10]
err_dq_list = [2, 6, 10, 14, 16]
err_pr_list = [2, 4, 6, 8, 10]

x = range(len(label_list))

error_params = dict(elinewidth=2, ecolor='black', capsize=3)  # 设置误差标记参数

fig, ax = plt.subplots()

rects1 = plt.bar(x, height=fedavg_list/fedavg_choose_list, width=0.15, color='#2ca02c', label="FedAvg [Google Team]",
hatch='.')
rects2 = plt.bar([i + 0.15 for i in x], height=fedcs_list/fedcs_choose_list, color='#1f77b4', width=0.15, label="FedCS [Nishio, 2019]",
hatch='xx')
rects3 = plt.bar([i + 0.3 for i in x], height=ddqn_list/pred_choose_list, color='#ff7f0e', width=0.15, label="DDQN-based (Proposed)",
hatch='')
rects4 = plt.bar([i + 0.45 for i in x], height=offline_list/offline_choose_list, color='#d62728', width=0.15,
label="Offline", hatch='\\')


plt.ylabel("Ratio of Valid Participants", fontsize=20)
plt.ylim(0, 1.6)
plt.yticks([0, 0.5, 1])

plt.xticks([index + 0.22 for index in x], label_list)

ax = plt.gca()  # 获取当前图像的坐标轴信息
ax.yaxis.get_major_formatter().set_powerlimits((0, 1))

plt.xlabel("FL Rounds", fontsize=21)
plt.tick_params(labelsize=18)
leg = ax.legend(fontsize=15)  # , frameon=False)
leg.set_draggable(True)
plt.tight_layout()

plt.show()