Trier une partie de matrice 2d sans utilser numpy

Version imprimable

Bonjour.
j'ai une matrice MxN d'entiers. Je veux ne trier que la sous-matrice commençant en (i,i) (ne pas prendre en compte les i premières lignes et colonnes) :?
Le tri doit s'effectuer sur la i° colonne par ordre décroissant.
par exemple pour i = 2 :

Code:

sort_sub_matrix([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]],2)

je veux obtenir :
[1, 1, 1, 1]
[2, 2, 2, 2]
[3, 3, 4, 4]
[4, 4, 3, 3]

J'ai codé cette fonction qui fournit le résultat attendu mais je trouve ça bien lourd :
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 def sort_sub_matrix(M, indice): M2 = [] for r in M[indice:]: M2.append(r[indice:]) M2.sort(reverse=True) for r_item, r_value in enumerate(M2): for c_item, c_value in enumerate(r_value): M[r_item + indice][c_item + indice] = c_value return M M = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]] for r in M: print(r) print('') M = sort_sub_matrix(M, 2) for r in M: print(r)
Bien sûr, c'est sans numpy sinon un coup de slicing et c'est terminé.
Si quelqu'un a une solution plus élégante, je suis preneur :hola:

Salut !

Peut-être comme ça ?

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
def sort_sub_matrix(M, indice):
 
    M2 = []
    for r in M[indice:]:
        M2.append(r[indice:])        
    M2.sort(reverse=True)    
 
    for i, row in enumerate(M[indice:]):
        M[indice+i] = row[:indice]+M2[i]
 
    return M
 
 
M = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]]
 
 
for r in M:
    print(r)
print('')
 
M = sort_sub_matrix(M, 1)
for r in M:
    print(r)

Voire comme ça, si on comprend le fonctionnement des listes ^^ :

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
def sort_sub_matrix(indice):
 
    M2 = []
    for r in M[indice:]:
        M2.append(r[indice:])        
    M2.sort(reverse=True)    
 
    for i, row in enumerate(M[indice:]):
        M[indice+i] = row[:indice]+M2[i]
 
 
 
M = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]]
 
 
for r in M:
    print(r)
print('')
 
sort_sub_matrix(1)
for r in M:
    print(r)

Salut,

Pour le fun:

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
>>> M = [[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]]
>>> i = 2
>>> z = [ m [i:] for m in M[i:]]
>>> z.sort(reverse=True)
>>> for j, m in enumerate(M[i:]): m [i:] = z[j]
...
>>> for e in M: print(e)
...
[1, 1, 1, 1]
[2, 2, 2, 2]
[3, 3, 4, 4]
[4, 4, 3, 3]
>>>

mais je ne suis pas sur que le "tri" soit "bon" mais il est "bug compatible" avec l'original.

- W

18/07/2022, 21h45
ypcman

Merci à tous les 2.
Le 1° code de Lenarvalo fonctionne parfaitement avec une boucle for en moins que mon code. En revanche, je ne vois pas l’intérêt du 2° code si on veut pouvoir appliquer la fct de tri à plusieurs matrices ...
~~Le code de wiztricks ne fonctionne pas pour un autre indice que 2 et même pour l'indice 2 avec une matrice plus quelconque, mais ton code n'est peut-être que de l'humour pythonesque~~
Finalement son code fonctionne aussi ...
18/07/2022, 22h57
Invité

Citation:

En revanche, je ne vois pas l’intérêt du 2° code si on veut pouvoir appliquer la fct de tri à plusieurs matrices ...

Tu peux enlever le return dans ce cas puisqu'il s'agit de listes (l'emplacement mémoire, tout ça, tout ça, j'ai pas le vocabulaire pour expliquer)...

Citation:

Envoyé par ypcman

Le code de wiztricks ne fonctionne pas pour un autre indice que 2 et même pour l'indice 2 avec une matrice plus quelconque, mais ton code n'est peut-être que de l'humour pythonesque

Pour moi, si je teste, çà fonctionne:
Code:

1 2 3 4 5 6 7 8 >>> M = [[1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]] >>> i = 1 >>> z = sorted([ m[i:] for m in M[i:]], reverse=True) >>> for j, m in enumerate(M[i:]): m [i:] = z[j] ... >>> M [[1, 1, 1], [2, 4, 4], [3, 3, 3], [4, 2, 2]] >>>
- W

Citation:
Envoyé par wiztricks
Pour moi, si je teste, çà fonctionne:
Code:

1 2 3 4 5 6 7 8 >>> M = [[1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]] >>> i = 1 >>> z = sorted([ m[i:] for m in M[i:]], reverse=True) >>> for j, m in enumerate(M[i:]): m [i:] = z[j] ... >>> M [[1, 1, 1], [2, 4, 4], [3, 3, 3], [4, 2, 2]] >>>
- W
Oui tout à fait ! je m'étais trompé en testant ton code qui est remarquablement synthétique