numpy.unique
Cette fonction renvoie un tableau d'éléments uniques dans le tableau d'entrée. La fonction peut renvoyer un tuple de tableau de valeurs uniques et un tableau d'indices associés. La nature des indices dépend du type de paramètre de retour dans l'appel de fonction.
numpy.unique(arr, return_index, return_inverse, return_counts)
Où,
N ° Sr. | Paramètre et description |
---|---|
1 | arr Le tableau d'entrée. Sera aplati si ce n'est pas un tableau 1-D |
2 | return_index Si True, retourne les indices des éléments dans le tableau d'entrée |
3 | return_inverse Si True, renvoie les indices du tableau unique, qui peuvent être utilisés pour reconstruire le tableau d'entrée |
4 | return_counts Si True, renvoie le nombre de fois où l'élément du tableau unique apparaît dans le tableau d'origine |
Exemple
import numpy as np
a = np.array([5,2,6,2,7,5,6,8,2,9])
print 'First array:'
print a
print '\n'
print 'Unique values of first array:'
u = np.unique(a)
print u
print '\n'
print 'Unique array and Indices array:'
u,indices = np.unique(a, return_index = True)
print indices
print '\n'
print 'We can see each number corresponds to index in original array:'
print a
print '\n'
print 'Indices of unique array:'
u,indices = np.unique(a,return_inverse = True)
print u
print '\n'
print 'Indices are:'
print indices
print '\n'
print 'Reconstruct the original array using indices:'
print u[indices]
print '\n'
print 'Return the count of repetitions of unique elements:'
u,indices = np.unique(a,return_counts = True)
print u
print indices
Sa sortie est la suivante -
First array:
[5 2 6 2 7 5 6 8 2 9]
Unique values of first array:
[2 5 6 7 8 9]
Unique array and Indices array:
[1 0 2 4 7 9]
We can see each number corresponds to index in original array:
[5 2 6 2 7 5 6 8 2 9]
Indices of unique array:
[2 5 6 7 8 9]
Indices are:
[1 0 2 0 3 1 2 4 0 5]
Reconstruct the original array using indices:
[5 2 6 2 7 5 6 8 2 9]
Return the count of repetitions of unique elements:
[2 5 6 7 8 9]
[3 2 2 1 1 1]