二栏二栏第三栏熊猫

Question

我有一个数据文件，如下所示，我希望通过分组模型和调度器列来绘制多个栏。例如，首先，多个bar属于ecaresnet50t三种不同的调度程序，表示mae评分。第二，三个多条表示resnest50d调度器的mae，等等。

         model     scheduler    mae
0   ecaresnet50t      warm      4.518
1   ecaresnet50t      cosine    4.46
2   ecaresnet50t      constant  4.972
3   resnest50d        warm      4.056
4   resnest50d        cosine    4.1
5   resnest50d        constant  5.072
6   resnetrs50        warm      4.164
7   resnetrs50        cosine    4.154
8   resnetrs50        constant  4.644
9   seresnet50        warm      4.202

我试过一些像(df.groupby(['model','scheduler'])['mae'].plot.bar())这样的东西，但它不起作用

Answer 1

使用枢轴台。它们是您要寻找的东西，在这种情况下比多索引组更容易使用。

df_pivot = pd.pivot_table(df, 
                          values="mae", 
                          index="model", 
                          columns="scheduler", 
                          aggfunc=np.mean)

df_pivot.plot.bar()

​

​