pandas idxmax

# By default, it returns the index for the maximum value in each column.
df.idxmax()
D Goglia