效果:
代码:
import matplotlib.pyplot as plt
similarity = [0.651, 0.802, 0.86] # similarity of action
divergence = [0.21748494048186232, 0.012685957834863208, 0.0027021711873171966] # js diversity
labels = ['Random PPO', 'Replicating model', 'Baseline']
plt.rcParams['axes.labelsize'] = 16 # xy轴label的size
plt.rcParams['xtick.labelsize'] = 12 # x轴ticks的size
plt.rcParams['ytick.labelsize'] = 14 # y轴ticks的size
# plt.rcParams['legend.fontsize'] = 12 # 图例的size
# 设置柱形的间隔
width = 0.3 # 柱形的宽度
x1_list = []
x2_list = []
for i in range(len(similarity)):
x1_list.append(i)
x2_list.append(i + width)
# 创建图层
fig, ax1 = plt.subplots()
# 设置左侧Y轴对应的figure
ax1.set_ylabel('Behavior Similarity')
ax1.set_ylim