Compter les valeurs uniques dans le tableau NumPy

Ce didacticiel présentera comment compter les occurrences de valeurs uniques dans un tableau NumPy.

Compter les valeurs uniques dans le tableau NumPy avec la fonction numpy.unique()

Pour compter le nombre d’occurrences de chaque élément unique dans le tableau numpy, nous pouvons utiliser la fonction numpy.unique(). Il prend le tableau comme argument d’entrée et renvoie tous les éléments uniques à l’intérieur du tableau dans l’ordre croissant. Nous pouvons spécifier le paramètre return_counts comme True pour obtenir également le nombre de fois que chaque élément est répété à l’intérieur du tableau. Voir l’exemple de code suivant.

import numpy as np

array = np.array([1, 1, 1, 2, 3, 4, 4, 4])

unique, counts = np.unique(array, return_counts=True)

result = np.column_stack((unique, counts))
print(result)

Production:

[[1 3]
 [2 1]
 [3 1]
 [4 3]]

Nous avons d’abord créé un tableau NumPy avec la fonction np.array(). Nous avons ensuite stocké tous les éléments uniques du tableau à l’intérieur du tableau unique et leur nombre d’occurrences respectif à l’intérieur du tableau comptes avec la fonction np.unique(). Nous avons ensuite compressé les deux tableaux 1D unique et counts dans un seul tableau 2D result avec la fonction np.column_stack().