Python numpy.argmax()

1. Syntaxe de numpy.argmax() :
2. Exemples de codes : numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau
3. Exemples de codes : Définissez le paramètre axis dans numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau
4. Exemples de codes : Définissez le paramètre out dans numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau

Python Numpy numpy.argmax() retourne les indices des valeurs les plus élevées dans le tableau NumPy donné.

Syntaxe de numpy.argmax() :

numpy.argmax(a, axis=None, out=None)

Paramètres

Renvoie

Un tableau d’indices d’éléments ayant la valeur la plus élevée de tout le tableau.

Exemples de codes : numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau

Trouver l’indice de la valeur la plus élevée dans un tableau 1-D

import numpy as np

a=np.array([2,6,1,5])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

print("\nThe largest value in the array:")
print(a[req_index])

Production:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Elle donne l’indice de l’élément ayant la plus grande valeur dans le tableau d’entrée donné.

Nous pouvons également trouver la plus grande valeur dans le tableau en utilisant l’index retourné par la méthode np.argmax().

Si nous avons deux plus grandes valeurs dans le tableau, la méthode retourne l’index du plus grand élément qui apparaît en premier dans le tableau.

import numpy as np

a=np.array([2,6,1,6])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

print("\nThe largest value in the array:")
print(a[req_index])

Production:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Ici, 6 est la plus grande valeur qui apparaît deux fois dans le tableau, mais la méthode np.argmax() retourne l’index de 6 à l’index 1 puisqu’il apparaît en premier dans le tableau.

Trouver l’indice de la valeur la plus élevée dans un tableau 2-D

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

Production:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Ici, comme le paramètre axis n’est pas donné, le tableau est aplati et l’indice du plus grand élément du tableau aplati est renvoyé.

Exemples de codes : Définissez le paramètre axis dans numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau

Trouver les indices de l’élément le plus élevé le long de l’axe des colonnes

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a,axis=0)
print("\nIndices with the largest value along column axis:")
print(req_index)

Production:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
[1 1 0]

Si nous mettons axis=0, cela donne les indices de la valeur la plus élevée le long de chaque colonne. Ici, la première colonne a la valeur la plus élevée à l’indice 1, la deuxième colonne a la valeur la plus élevée à l’indice 1 et la troisième colonne a la valeur la plus élevée à l’indice 0.

Trouver les indices des éléments les plus élevés le long de l’axe des lignes

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a,axis=1)
print("\nIndices with the largest value along row axis:")
print(req_index)

Production:

Array:
[[2 1 6]
 [7 4 5]]

Indices with the largest value along the row axis:
[2 0]

Si nous fixons axis=1, cela donne les indices des valeurs les plus élevées le long de chaque ligne.

Exemples de codes : Définissez le paramètre out dans numpy.argmax() Méthode pour trouver les indices des plus grandes valeurs dans un tableau

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

req_index=np.array(0)

print("Array:")
print(a)

np.argmax(a,out=req_index)
print("\nIndex with the largest value:")
print(req_index)

Production:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Ici, la variable req_index agit comme un paramètre de remplacement pour la valeur requise de l’indice et nous devons nous assurer que la taille de la sortie doit être égale à req_index.