Як використовувати функцію idxmax() у pandas (з прикладами)

Ви можете використовувати функцію pandas.DataFrame.idxmax() , щоб повернути індекс максимального значення на вказаній осі в pandas DataFrame.

Ця функція використовує такий синтаксис:

DataFrame.idxmax(axis=0, skipna=True)

золото:

• axis : вісь для використання (0 = рядки, 1 = стовпці). Значення за замовчуванням 0.
• skipna : чи виключати NA або нульові значення. Типовим значенням є True.

Наступні приклади показують, як використовувати цю функцію на практиці з такими pandas DataFrame:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({'points': [25, 12, 15, 8, 9, 23],
                   'assists': [5, 7, 7, 9, 12, 9],
                   'rebounds': [11, 8, 11, 6, 6, 5]},
                   index=['Andy','Bob', 'Chad', 'Dan', 'Eric', 'Frank'])

#view DataFrame
df

        points assists rebounds
Andy 25 5 11
Bob 12 7 8
Chad 15 7 11
Dan 8 9 6
Eric 9 12 6
Frank 23 9 5

Приклад 1: знайдіть індекс із максимальним значенням для кожного стовпця

Наступний код показує, як знайти індекс, який має максимальне значення для кожного стовпця:

 #find index that has max value for each column
df. idxmax (axis= 0 )

Andy points
assists Eric
rebounds Andy
dtype:object

З результату ми бачимо:

• Гравець із найвищим значенням у колонці очок – Енді.
• Гравець з найвищим значенням у колонці асистів – це Ерік.
• Гравець із найвищим значенням у колонці підбирань – Енді.

Важливо зауважити, що функція idxmax() поверне перше повторення максимального значення.

Наприклад, зауважте, що Енді та Чед зробили по 11 підбирань. Оскільки Енді з’являється першим у DataFrame, повертається його ім’я.

Приклад 2: знайдіть стовпець із максимальним значенням для кожного рядка

Наступний код показує, як знайти стовпець, що містить максимальне значення для кожного рядка:

 #find column that has max value for each row
df. idxmax (axis= 1 )

Andy points
Bob stitches
Chad points
Dan assists
Eric assists
Frank points
dtype:object

З результату ми бачимо:

• Найвище значення в рядку з позначкою “Енді” знаходиться в стовпці балів .
• Найвище значення в рядку з позначкою “Боб” знаходиться в стовпці балів .
• Найвище значення в рядку з позначкою «Чад» знаходиться в стовпці балів .
• Найвище значення в рядку з позначкою «Dan» знаходиться в колонці передач .
• Найвище значення в рядку з позначкою «Ерік» знаходиться в колонці передач .
• Найвище значення в рядку з позначкою “Енді” знаходиться в стовпці балів .

Зверніться до документації pandas для повного пояснення функції idxmax().

Додаткові ресурси

Як знайти максимальне значення стовпців у Pandas
Як обчислити середнє значення стовпців у Pandas
Як обчислити суму стовпців у Pandas