Numpy NP.argmax
Numpy vous fournit la fonction argmax () qui vous permet de récupérer l'index de l'élément maximum dans un tableau sur un axe spécifique.
Ce tutoriel tente d'expliquer comment le Numpy.La fonction argmax () fonctionne et comment l'utiliser.
Syntaxe de fonction
La syntaxe de la fonction est comme indiqué ci-dessous:
nombant.argmax (tableau, axe = aucun, out = aucun, *, keepdims =La fonction accepte les paramètres suivants:
- tableau - fait référence au tableau d'entrée sur lequel la fonction est appliquée.
- Axe-spécifie le long de lequel l'axe du tableau la fonction est appliquée - il s'agit d'un paramètre de type entier facultatif. La fonction aplatira le tableau si la valeur de l'axe n'est pas définie.
- Out - Spécifie un tableau dans lequel le résultat est inséré. Le tableau spécifié doit être du type et de la forme appropriés.
- KeepDims - Il s'agit d'un paramètre booléen qui prend de vrai ou de faux. S'il est réglé sur true, l'axe réduit est laissé sous forme de dimensions avec une taille un.
Valeur de retour de fonction
La fonction renvoie un tableau d'indices dans le tableau avec la même forme que le tableau.forme.
Exemple
Illustrons comment utiliser le Numpy.fonction argmax avec un exemple.
Commencez par importer Numpy comme:
# Importy NumpyImporter Numpy comme NP
Ensuite, créez un tableau 2D comme indiqué:
# Tableau 2Darray = np.Arange (6).Reshape ((2,3))
Imprimer (tableau)
Ce qui précède doit créer un tableau 2D avec les éléments comme indiqué:
[[0 1 2][3 4 5]]
Pour utiliser la fonction argmax () sur l'ensemble du tableau (aplati), nous pouvons exécuter le code comme indiqué:
# élément maximalprint (f "Valeur max: np.argmax (array, axe = non) ");
Le code ci-dessus renvoie l'index maximum de la valeur maximale comme indiqué:
Vous pouvez également utiliser la fonction argmax le long d'un axe spécifique comme indiqué:
# élément maximum le long de l'axe 0Imprimer (NP.argmax (tableau, axe = 0));
Remarque: Puisque nous travaillons avec un tableau 2D, le tableau a deux axes possibles. Le premier passe verticalement à travers les lignes sous forme d'axe 0, et l'autre passe horizontalement à travers les colonnes sous forme d'axe 1.
Par conséquent, le code ci-dessus doit revenir:
Le même cas s'applique le long de l'axe 1. Un exemple est comme indiqué:
# élément max le long de l'axe 1Imprimer (NP.argmax (tableau, axe = 1))
Cela devrait revenir:
[2 2]Exemple 2
Nous pouvons également utiliser la fonction argmax avec un tableau de dimension à n. Par exemple, le code ci-dessous illustre comment utiliser la fonction avec un tableau 3D.
# Tableau 3Darray = np.Arange (24).Reshape (2, 3, 4)
Imprimer (tableau)
Cela devrait générer un tableau comme:
[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
Le long des axes, nous pouvons obtenir l'argmax comme indiqué:
# élément maximalprint (f "Max: np.argmax (tableau, axe = aucun) ")
# élément maximal le long de l'axe 0
imprimer (f "Indices max: np.argmax (tableau, axe = 0) ")
# élément max le long de l'axe 1
imprimer (f "Indices max: np.argmax (tableau, axe = 1) ")
# élément max le long de l'axe 2
imprimer (f "Indices max: np.argmax (tableau, axe = 2) ")
Le code ci-dessus doit retourner:
Max: 23Indices max: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
Indices max: [[2 2 2 2]
[2 2 2 2]]
Indices max: [[3 3 3]
[3 3 3]]
Fermeture
Cet article explore comment utiliser la fonction Numpy Argmax () pour récupérer les indices des valeurs max avec un index spécifique.
Merci d'avoir lu et restez à l'écoute!!