My Pandas DataFrame, df, looks like this:
parameter1 parameter2 value
1 1 0.1
2 0.2
2 1 0.6
2 0.3
value is the result of a groupby(['parameter1','parameter2']).mean() on another DataFrame. Now, I can find the maximum value of value for each value of parameter1 using
df.max(level='parameter1')
However, I need to find the corresponding value of parameter2 for this maximum value. It seems df.idxmax() does not support level=, so how can I do this instead?
A nice way would be
df.unstack().idxmax(axis=1)
Unstacking the dataframe gives a dataframe with parameter_1 as the column index.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With