Calculer le mode de tableau dans NumPy

1. Calculer le mode d’un tableau NumPy avec la fonction scipy.stats.mode()
2. Calculer le mode d’un tableau NumPy avec la fonction numpy.unique()

Ce tutoriel présentera comment calculer le mode d’un tableau NumPy en Python.

Calculer le mode d’un tableau NumPy avec la fonction scipy.stats.mode()

Le mode est la valeur la plus répétée dans une collection. La bibliothèque scipy.stats contient de nombreuses fonctions liées aux statistiques. La fonction mode() à l’intérieur de la bibliothèque scipy.stats trouve le mode d’un tableau en Python. Il prend un tableau comme argument d’entrée et renvoie un tableau des valeurs les plus courantes à l’intérieur du tableau d’entrée. Pour que cette méthode fonctionne, nous devons installer le package scipy. La commande pour l’installer est donnée ci-dessous.

pip install scipy

L’exemple de code suivant nous montre comment calculer le mode à l’intérieur d’un tableau NumPy avec la fonction scipy.stats.mode().

import numpy as np
from scipy import stats

array = np.array([1, 2, 3, 4, 4, 5])
mode = stats.mode(array)
print(mode[0])

Production:

[4]

Nous avons d’abord créé le tableau array avec la fonction np.array(). Nous avons ensuite calculé le mode avec la fonction scipy.stats.mode() et stocké le résultat dans le tableau mode. Au final, nous avons affiché la valeur la plus répétée en imprimant le premier élément du tableau mode.

Calculer le mode d’un tableau NumPy avec la fonction numpy.unique()

Si nous voulons utiliser le package NumPy uniquement pour trouver le mode, nous pouvons utiliser la fonction numpy.unique(). La fonction numpy.unique() prend un tableau comme argument d’entrée et renvoie un tableau de tous les éléments uniques à l’intérieur du tableau d’entrée. Nous pouvons également spécifier le paramètre return_count sur True pour obtenir le nombre de fois où chaque élément unique est répété à l’intérieur du tableau d’entrée.

import numpy as np

array = np.array([1, 2, 3, 4, 4, 5])
vals, counts = np.unique(array, return_counts=True)
index = np.argmax(counts)
print(vals[index])

Production:

4

Dans le code ci-dessus, nous avons calculé le mode du tableau NumPy array avec les fonctions np.unique() et np.argmax() en Python. Nous avons d’abord créé le tableau array avec la fonction np.array(). Nous avons ensuite utilisé la fonction np.unique() et stocké les valeurs uniques dans le tableau vals, ainsi que le nombre de fois où chaque valeur est répétée dans le tableau counts. Nous avons ensuite calculé la valeur maximale à l’intérieur du tableau counts avec la fonction np.argmax() et stocké la valeur à l’intérieur de la variable index. Au final, nous avons affiché le mode en imprimant la valeur à l’index index du tableau vals.