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