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