Numpy NP.Argmax
Numpy zapewnia funkcję argmax (), która pozwala pobrać indeks elementu maksymalnego w tablicy na określonej osi.
Ten samouczek próbuje wyjaśnić, w jaki sposób Numpy.funkcja argmax () działa i jak z niej korzystać.
Składnia funkcji
Składnia funkcji jest jak pokazano poniżej:
Numpy.argmax (tablica, axis = Brak, out = brak, *, klepedims =Funkcja akceptuje następujące parametry:
- tablica - odnosi się do tablicy wejściowej, na której zastosowano funkcję.
- Oś -Okazuje się, wzdłuż której osi tablicy jest stosowana -to jest opcjonalnym parametrem typu liczb całkowitych. Funkcja spłaszczy tablicę, jeśli wartość osi nie jest ustawiona.
- OUT - określa tablicę, do której wkłada się wynik. Określona tablica powinna mieć odpowiedni typ i kształt.
- Keepdims - jest to parametr logiczny, który przyjmuje prawdziwe lub fałszywe. Jeśli jest ustawiona na true, zmniejszona oś jest pozostawiona jako wymiary o rozmiar.
Funkcja zwraca wartość
Funkcja zwraca tablicę wskaźników do tablicy o tym samym kształcie co tablica.kształt.
Przykład
Zilustrujmy, jak używać Numpy.Funkcja argmax z przykładem.
Zacznij od importu Numpy jako:
# Importsy Numpyimportować Numpy jako NP
Następnie utwórz tablicę 2D, jak pokazano:
# 2d tablicaarray = np.Arange (6).Reshape ((2,3))
Drukuj (tablica)
Powyższe powinno utworzyć tablicę 2D z elementami, jak pokazano:
[[0 1 2][3 4 5]]
Aby użyć funkcji argmax () w całej tablicy (spłaszczona), możemy uruchomić kod, jak pokazano:
# element maksymalnydrukuj (f "max wartość: np.argmax (tablica, axis = none) ");
Powyższy kod zwraca indeks maksymalny wartości maksymalnej, jak pokazano:
Możesz także użyć funkcji Argmax wzdłuż określonej osi, jak pokazano:
# element maksymalny wzdłuż osi 0Drukuj (NP.argmax (tablica, oś = 0));
Uwaga: Ponieważ pracujemy z tablicą 2D, tablica ma dwie możliwe osie. Pierwszy przebiegaj pionowo po rzędach jako oś 0, a drugi biegnie poziomo po kolumnach jako oś 1.
Stąd powyższy kod powinien zwrócić:
Ten sam przypadek dotyczy osi 1. Przykładem jest jak pokazano:
# element maksymalny wzdłuż osi 1Drukuj (NP.argmax (tablica, oś = 1))
To powinno powrócić:
[2 2]Przykład 2
Możemy również użyć funkcji Argmax z tablicą N-wymiaru. Na przykład poniższy kod ilustruje, jak korzystać z funkcji z tablicą 3-D.
# 3D tablicaarray = np.Arange (24).Reshape (2, 3, 4)
Drukuj (tablica)
Powinno to wygenerować tablicę jako:
[[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
Wzdłuż osi możemy uzyskać Argmax, jak pokazano:
# element maksymalnydruk (f "max: np.argmax (tablica, axis = none) ")
# element maksymalny wzdłuż osi 0
drukuj (f "max wskaźniki: np.argmax (tablica, axis = 0) ")
# element maksymalny wzdłuż osi 1
drukuj (f "max wskaźniki: np.argmax (tablica, axis = 1) ")
# element maksymalny wzdłuż osi 2
drukuj (f "max wskaźniki: np.argmax (tablica, axis = 2) ")
Kod powyżej powinien zwrócić:
Max: 23MAX Wskaźniki: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
MAX Wskaźniki: [[2 2 2 2]
[2 2 2 2]]
MAX Wskaźniki: [[3 3 3]
[3 3 3]]
Zamknięcie
W tym artykule bada sposób korzystania z funkcji Numpy Argmax () do pobrania wskaźników wartości maksymalnej wraz z określonym indeksem.
Dziękuję za przeczytanie i bądźcie czujni!!