Comment puis-je trier un tableau NumPy par une colonne spécifique ?

Tri d'un tableau NumPy par colonne

Tri d'un tableau NumPy par sa colonne spécifique peut souvent être une opération nécessaire pour la manipulation de données. Par exemple, considérons le tableau :

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

Supposons que nous souhaitions réorganiser les lignes d'un en fonction des valeurs de la deuxième colonne. Pour y parvenir, nous pouvons utiliser la fonction argsort() dans NumPy.

a[:, 1].argsort()

L'expression ci-dessus applique argsort() à la deuxième colonne de a, produisant un tableau d'indices qui correspondent aux valeurs triées. Plus précisément, il renvoie :

array([1, 0, 2])

Ces indices représentent l'ordre dans lequel les lignes de a doivent être réorganisées pour trier par la deuxième colonne par ordre croissant. En combinant cela avec l'indexation de tableau, nous pouvons obtenir le tableau trié :

a[a[:, 1].argsort()]

Cette expression renvoie :

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])

Comme souhaité, les lignes de a ont été triées selon la deuxième colonne .

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!