Pandas DataFrame idxmax() Method

Return the index of the row with the highest sales, and the index of the row with the highest age:

import pandas as pd

data = {
"sales": [23, 34, 56],
"age": [50, 40, 30]
}

df = pd.DataFrame(data)

print(df.idxmax())

Definition and Usage

The idxmax() method returns a Series with the index of the maximum value for each column.

By specifying the column axis (axis='columns'), the idxmax() method returns a Series with the index of the maximum value for each row.

dataframe.idxmax(axis, skipna)

The parameters are keyword arguments.

Parameter	Value	Description
axis	`0 1 'index' 'columns'`	Optional, Which axis to check, default 0.
skipna	`True False`	Optional, default True. Specifies whether to skip NULL values or not.

A Series with the indexes of the rows/columns with the highest values.