Python 与最大值对应的列名

我有数据帧

df

         0.0         1.0         2.0         3.0         4.0         5.0
0   0.537592    0.255088    0.102097    0.062989    0.024679    0.006685
1   0.063914    0.077319    0.173972    0.276244    0.121987    0.021594

我需要在每一排取最大值。所以第一行的最大值是

0.537592

[列名

0.0

]，第二行的最大值是

0.276244

[列名

3.0

]

所以我需要你喜欢的

        label        
0        0.0
1        3.0

我怎样才能做到这一点？ 尝试了类似于

subs.loc[subs['0.0']==subs['0.0'].max（axis=1）]

的方法，但没有任何帮助

您希望：

df.idxmax(axis=1).to_frame(name='Label')

   Label
0   0.0
1   3.0

您希望为此：

df.idxmax(axis=1).to_frame(name='Label')

   Label
0   0.0
1   3.0