Comment obtenir l’indice de valeur maximale dans le tableau NumPy

Vous pouvez utiliser les méthodes suivantes pour obtenir l’index de la valeur maximale dans un tableau NumPy :

Méthode 1 : obtenir l’indice de la valeur maximale dans un tableau unidimensionnel

x.argmax()

Méthode 2 : obtenir l’index de la valeur maximale dans chaque ligne du tableau multidimensionnel

x.argmax(axis=1)

Méthode 3 : obtenir l’index de la valeur maximale dans chaque colonne du tableau multidimensionnel

x.argmax(axis=0)

Les exemples suivants montrent comment utiliser chaque méthode dans la pratique.

Exemple 1 : obtenir l’indice de la valeur maximale dans un tableau unidimensionnel

Le code suivant montre comment obtenir l’index de la valeur maximale dans un tableau NumPy unidimensionnel :

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

La fonction argmax() renvoie une valeur de 2 .

Cela nous indique que la valeur en position d’index 2 du tableau contient la valeur maximale.

Si nous regardons le tableau d’origine, nous pouvons voir que la valeur en position d’index 2 est 9 , ce qui est en effet la valeur maximale du tableau.

Exemple 2 : obtenir l’index de la valeur maximale dans chaque ligne d’un tableau multidimensionnel

Le code suivant montre comment obtenir l’index de la valeur maximale dans chaque ligne d’un tableau NumPy multidimensionnel :

import numpy as np

#create multi-dimentsional 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)

D’après les résultats, nous pouvons voir :

• La valeur maximale de la première ligne est située en position d’index 3 .
• La valeur maximale de la deuxième ligne est située en position d’index 1 .

Exemple 3 : obtenir l’index de la valeur maximale dans chaque colonne d’un tableau multidimensionnel

Le code suivant montre comment obtenir l’index de la valeur maximale dans chaque colonne d’un tableau NumPy multidimensionnel :

import numpy as np

#create multi-dimentsional 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)

D’après les résultats, nous pouvons voir :

• La valeur maximale dans la première colonne est située en position d’index 1 .
• La valeur maximale dans la deuxième colonne est située en position d’index 1 .
• La valeur maximale dans la troisième colonne est située en position d’index 1 .
• La valeur maximale dans la quatrième colonne est située en position d’index 0 .

Connexes : Une explication simple des axes NumPy

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes en Python :

Comment remplir un tableau NumPy avec des valeurs
Comment remplacer des éléments dans un tableau NumPy
Comment obtenir une ligne spécifique à partir d’un tableau NumPy