pandas.core.groupby.DataFrameGroupBy.idxmax

DataFrameGroupBy.idxmax(axis=0, skipna=True, numeric_only=_NoDefault.no_default)[source]

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

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

Parameters

axis:{0 или ‘index’, 1 или ‘columns’}, по умолчанию 0

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

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

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

numeric_only:bool, по умолчанию True для axis=0, False для axis=1

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

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

Returns

Series

Индексы максимумов по указанной оси.

Raises

ValueError

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

См. также

Series.idxmax

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

Notes

Этот метод является версией для 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