Jak uzyskać indeks maksymalnej wartości w tablicy numpy

Możesz użyć następujących metod, aby uzyskać indeks maksymalnej wartości w tablicy NumPy:

Metoda 1: Uzyskaj indeks maksymalnej wartości w tablicy jednowymiarowej

 x. argmax ()

Metoda 2: Uzyskaj indeks maksymalnej wartości w każdym wierszu tablicy wielowymiarowej

 x. argmax (axis= 1 )

Metoda 3: Uzyskaj indeks maksymalnej wartości w każdej kolumnie tablicy wielowymiarowej

 x. argmax (axis= 0 )

Poniższe przykłady pokazują, jak zastosować każdą metodę w praktyce.

Przykład 1: Uzyskaj indeks maksymalnej wartości w tablicy jednowymiarowej

Poniższy kod pokazuje, jak uzyskać indeks wartości maksymalnej w jednowymiarowej tablicy NumPy:

 import numpy as np

#create NumPy array of values
x = np. array ([2, 7, 9, 4, 4, 6, 3])

#find index that contains max value
x. argmax ()

2

Funkcja argmax() zwraca wartość 2 .

To mówi nam, że wartość na pozycji indeksu 2 tablicy zawiera wartość maksymalną.

Jeśli spojrzymy na oryginalną tablicę, zobaczymy, że wartość na pozycji indeksu 2 wynosi 9 , co w rzeczywistości jest maksymalną wartością w tablicy.

Przykład 2: Uzyskaj indeks maksymalnej wartości w każdym wierszu tablicy wielowymiarowej

Poniższy kod pokazuje, jak uzyskać indeks maksymalnej wartości w każdym wierszu wielowymiarowej tablicy NumPy:

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each row
x. argmax (axis= 1 )

array([3, 1], dtype=int32)

Z wyników możemy zobaczyć:

• Maksymalna wartość pierwszego wiersza znajduje się na pozycji indeksu 3 .
• Maksymalna wartość drugiej linii znajduje się na pozycji indeksu 1 .

Przykład 3: Uzyskaj indeks maksymalnej wartości w każdej kolumnie tablicy wielowymiarowej

Poniższy kod pokazuje, jak uzyskać indeks maksymalnej wartości w każdej kolumnie wielowymiarowej tablicy NumPy:

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each column
x. argmax (axis= 0 )

array([1, 1, 1, 0], dtype=int32)

Z wyników możemy zobaczyć:

• Maksymalna wartość w pierwszej kolumnie znajduje się na pozycji indeksu 1 .
• Maksymalna wartość w drugiej kolumnie znajduje się na pozycji indeksu 1 .
• Maksymalna wartość w trzeciej kolumnie znajduje się na pozycji indeksu 1 .
• Maksymalna wartość w czwartej kolumnie znajduje się na pozycji indeksu 0 .

Powiązane: Proste wyjaśnienie osi NumPy

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje w Pythonie:

Jak wypełnić tablicę NumPy wartościami
Jak zamienić elementy w tablicy NumPy
Jak uzyskać określony wiersz z tablicy NumPy