Comment calculer le mode du tableau NumPy (avec exemples)

Vous pouvez utiliser la syntaxe de base suivante pour trouver le mode d’un tableau NumPy :

#find unique values in array along with their counts
vals, counts = np.unique(array_name, return_counts=True)

#find mode
mode_value = np.argwhere(counts == np.max(counts))

Rappelons que le mode est la valeur qui apparaît le plus souvent dans un tableau.

Notez qu’il est possible qu’un tableau ait un ou plusieurs modes.

Les exemples suivants montrent comment utiliser cette syntaxe dans la pratique.

Exemple 1 : Calcul du mode de tableau NumPy avec un seul mode

Le code suivant montre comment trouver le mode d’un tableau NumPy dans lequel il n’y a qu’un seul mode :

import numpy as np

#create NumPy array of values with only one mode
x = np.array([2, 2, 2, 3, 4, 4, 5, 5, 5, 5, 7])

#find unique values in array along with their counts
vals, counts = np.unique(x, return_counts=True)

#find mode
mode_value = np.argwhere(counts == np.max(counts))

#print list of modes
print(vals[mode_value].flatten().tolist())

[5]

#find how often mode occurs
print(np.max(counts))

4

À partir de la sortie, nous pouvons voir que le mode est 5 et qu’il se produit 4 fois dans le tableau NumPy.

Exemple 2 : Mode de calcul du tableau NumPy avec plusieurs modes

Le code suivant montre comment trouver le mode d’un tableau NumPy dans lequel il existe plusieurs modes :

import numpy as np

#create NumPy array of values with multiple modes
x = np.array([2, 2, 2, 3, 4, 4, 4, 5, 5, 5, 7])

#find unique values in array along with their counts
vals, counts = np.unique(x, return_counts=True)

#find mode
mode_value = np.argwhere(counts == np.max(counts))

#print list of modes
print(vals[mode_value].flatten().tolist())

[2, 4, 5]

#find how often mode occurs
print(np.max(counts))

3

D’après la sortie, nous pouvons voir que ce tableau NumPy a trois modes : 2 , 4 et 5 .

On peut également voir que chacune de ces valeurs apparaît 3 fois dans le tableau.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans NumPy :

Comment mapper une fonction sur un tableau NumPy
Comment trouver l’indice de valeur dans le tableau NumPy
Comment calculer la magnitude d’un vecteur à l’aide de NumPy