tri de matrices suivant un indice

Bonjour,

Un problème qui a déjà du être traité de nombreuses fois, mais je ne trouve pas de reference !

je voudrais trier une matrice suivant une ligne ou une colonne.

j'ai une solution mais quoique compacte, je la trouve un peu compliquée , je voulais savoir s'il n'y avait pas une fonction numpy qui fasse le travail directement, et faire la même chose sans transposer la matrice.

Code:

---

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37

# coding: utf-8   import numpy as np   l1=np.array([ 3, 4, 5, 8, 9, 2,6,7]) l2=np.array([30, 40, 5, 80, 90, 20,60,7]) l3=np.array([300, 400, 50, 800, 900, 200,60,70])   c=np.stack((l1,l2,l3))       print('matrice de base') print (c)     # tri   c=c.T print('matrice transposée') print(c)   d=c[c[:,0].argsort()]   print('tri sur 1ère colonne') print(d)   e=c[c[:,1].argsort()]   print('tri sur 2ème colonne') print(e)   f=c[c[:,2].argsort()]   print('tri sur 3ème colonne') print(f)

---

Salut,

Citation:

---

Envoyé par flyingfr

je voudrais trier une matrice suivant une ligne ou une colonne.

---

Pour trier suivant une colonne, vous avez l'argument axis qui permet de préciser laquelle.

Pour trier suivant une ligne sans transposer, vous pouvez récupérer les index d'argsort:

Code:

---

1 2 3 4 5 6 7

>>> ar = np.array([ [ 3, 4, 5, 8, 9, 2,6,7], ... [ 30, 40, 5, 80, 90, 20, 60, 7], ... [300, 400, 50, 800, 900, 200, 60, 70]]) >>> >>> idx = np.argsort(ar[0]) >>> idx array([5, 0, 1, 2, 6, 7, 3, 4], dtype=int64)

---

et construire le tableau trié:

Code:

---

1 2 3 4 5

>>> ar[:,idx] array([[ 2, 3, 4, 5, 6, 7, 8, 9], [ 20, 30, 40, 5, 60, 7, 80, 90], [200, 300, 400, 50, 60, 70, 800, 900]]) >>>

---

- W

Merci,

En fait pour trier une matrice a en lignes suivant la ligne i je fais

Code:

---

1 2	g=a[:,np.argsort(a[i])] # tri sur la ligne i de a

---

et pour trier une matrice en colonnes suivant une colonne

Code:

---

1 2	f=a[a[:,i].argsort()] # tri sur colonne i

---

Pour compléter le sujet, j'ai également trouvé que l'on pouvait faire de la manière suivante avec une liste en utilisant une fonction lambda

Code:

---

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

print('liste de tuples à trier') liste=[(7,19),(15,15),(4,12),(17,3),(6,8),(12,25),(2,3)] print(liste) lambda colonnes: colonnes[0] liste2=sorted(liste,key=lambda colonnes: colonnes[0])   print() print ('liste de tuples triée suivant premier terme') print(liste2) print() # liste à 2 dimensions liste3=[[1,10],[2,12],[12,18],[7,19],[15,15],[4,12],[17,3],[6,8],[12,25],[2,3]] print ( 'liste à trier',liste3) liste4=sorted(liste3,key=lambda colonnes: colonnes[1]) print ('liste à 2 dimensions triée sur le 2ème terme') print(liste4)

---