pandas.DataFrame.idxmax

DataFrame.idxmax(axis=0, skipna=True, numeric_only=False)[source]

Возвращает индекс первого вхождения максимального значения по указанной оси.

Значения NA/NULL исключаются.

Параметры:

axis:{0 или ‘индекс’, 1 или ‘колонки’}, по умолчанию 0

Ось для использования. 0 или ‘индекс’ для строк, 1 или ‘колонки’ для столбцов.

skipna:bool, по умолчанию True

Исключить значения NA/NULL. Если вся строка/столбец содержит NA, результат будет NA.

numeric_only:bool, по умолчанию False

Включать только данные типа float, int или boolean.

Добавлена в версии 1.5.0.

Возвращает:

Series

Индексы максимальных значений по указанной оси.

Возможные исключения:

ValueError

• Если строка/столбец пуст

См. также

Series.idxmax

Возвращает индекс максимального элемента.

Примечания

Этот метод — версия DataFrame для ndarray.argmax.

Примеры

Рассмотрим набор данных, содержащий потребление продуктов питания в Аргентине.

>>> df = pd.DataFrame({'consumption': [10.51, 103.11, 55.48],
...                     'co2_emissions': [37.2, 19.66, 1712]},
...                   index=['Pork', 'Wheat Products', 'Beef'])

>>> df
                consumption  co2_emissions
Pork                  10.51         37.20
Wheat Products       103.11         19.66
Beef                  55.48       1712.00

По умолчанию, он возвращает индекс максимального значения для каждого столбца.

>>> df.idxmax()
consumption     Wheat Products
co2_emissions             Beef
dtype: object

Для возвращения индекса максимального значения для каждой строки используйте axis="columns".

>>> df.idxmax(axis="columns")
Pork              co2_emissions
Wheat Products     consumption
Beef              co2_emissions
dtype: object