python - Pandas:每组的最大值索引

Question

我的 Pandas DataFrame，df，看起来像这样:

parameter1   parameter2   value

1            1            0.1
             2            0.2

2            1            0.6
             2            0.3

value 是另一个 DataFrame 上 groupby(['parameter1','parameter2']).mean() 的结果。现在，我可以使用 parameter1 的每个值找到 value 的最大值

df.max(level='parameter1')

但是，我需要找到这个最大值对应的parameter2的值。看来 df.idxmax() 不支持 level=，那么我该怎么做呢？

Answer 1

一个好的方法是

df.unstack().idxmax(axis=1)

对数据帧进行拆栈会得到一个以 parameter_1 作为列索引的数据帧。