我有一个数据框,我想绘制一个图例,包括“ A”、“ B”和“ C”,但是我的代码只生成了一个带有“ A”标签的图例:
data = {'A1_mean': [0.457, 1],
'A2_median': [0.391,1],
'A3_range': [0.645,1],
'A4_std': [0.111,1],
'B1_mean': [0.132,3],
'B2_median': [0.10,3],
'B3_range': [0.244,3],
'B4_std': [0.297,3],
'C1_mean': [0.286,2],
'C2_median': [0.231,2],
'C3_range': [0.554,2],
'C4_std': [0.147,2]}
df = pd.DataFrame(data).T
color = {1:'red',2:'green',3:'blue'}
ax=df[0].plot(kind='bar',color=df[1].map(color).tolist())
ax.legend(['A','B','C'])
提供:
我该如何修改它,以便我有一个带有A、B和C的图例,并且具有适当的颜色(A:红色,B:蓝色,C:绿色)?