Get your own Python server Result Size: 625 x 565
x
 
import pandas as pd
data = {
  "sales": [23, 34, 56],
  "age": [50, 40, 30]
}
df = pd.DataFrame(data)
print(df.idxmax())
  sales    2
  age      0
  dtype: int64