Discussion :

[flyingfr]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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)

[wiztricks]

Salut,

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[flyingfr]

Merci,

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

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[flyingfr]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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)