Jak korzystać z funkcji idxmax() w pandach (z przykładami)

Funkcji pandas.DataFrame.idxmax() można użyć do zwrócenia indeksu maksymalnej wartości na określonej osi w ramce DataFrame pandy.

Ta funkcja używa następującej składni:

DataFrame.idxmax(oś=0, pomiń=prawda)

Złoto:

• oś : oś, której należy użyć (0 = wiersze, 1 = kolumny). Wartość domyślna to 0.
• skipna : Określa, czy wykluczać wartości NA lub null. Wartość domyślna to Prawda.

Poniższe przykłady pokazują, jak w praktyce używać tej funkcji z następującą ramką DataFrame pand:

 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

Przykład 1: Znajdź indeks o maksymalnej wartości dla każdej kolumny

Poniższy kod pokazuje, jak znaleźć indeks mający maksymalną wartość dla każdej kolumny:

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

Andy points
assists Eric
rebounds Andy
dtype:object

Z wyniku możemy zobaczyć:

• Graczem z największą wartością w kolumnie punktów jest Andy.
• Graczem z najwyższą wartością w kolumnie asyst jest Eric.
• Zawodnikiem z największą wartością w kolumnie zbiórek jest Andy.

Należy zauważyć, że funkcja idxmax() zwróci pierwsze wystąpienie wartości maksymalnej.

Na przykład zauważ, że Andy i Chad mieli po 11 zbiórek. Ponieważ Andy pojawia się jako pierwszy w DataFrame, zwracane jest jego imię i nazwisko.

Przykład 2: Znajdź kolumnę z maksymalną wartością w każdym wierszu

Poniższy kod pokazuje, jak znaleźć kolumnę zawierającą maksymalną wartość dla każdego wiersza:

 #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

Z wyniku możemy zobaczyć:

• Najwyższa wartość w wierszu „Andy” znajduje się w kolumnie punktów .
• Najwyższa wartość w wierszu „Bob” znajduje się w kolumnie punktów .
• Najwyższa wartość w wierszu „Czad” znajduje się w kolumnie punktów .
• Najwyższa wartość w wierszu „Dan” znajduje się w kolumnie asyst .
• Najwyższa wartość w wierszu „Eric” znajduje się w kolumnie asyst .
• Najwyższa wartość w wierszu „Andy” znajduje się w kolumnie punktów .

Pełne wyjaśnienie funkcji idxmax() można znaleźć w dokumentacji pand .

Dodatkowe zasoby

Jak znaleźć maksymalną wartość kolumn w Pandach
Jak obliczyć średnią kolumn w Pandach
Jak obliczyć sumę kolumn w Pandach