Discussion :

[rosef]

bonjour,
je veux convertir une liste à une matrix en python.
j'ai une fonction qui renvoi une liste d'entier, moi je veux travailler sur une matrice, alors j'ai essayer de convertir la liste en matrice mais un message d'erreur apparait:"total size of new array must be unchanged".

par example ma fonction renvoi cette liste:
[1,4,7,2,5,8,3,6,9] alors mon objectif est que la matrice soit comme suit:
[[1,2,3],
[4,5,6],
[7,8,9]]

[tyrtamos]

Bonjour,

Une solution basique:

Code :

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

1
2
3
4
5
6
7
8
9
10
L = [1,4,7,2,5,8,3,6,9]
 
M = []
for i in xrange(0,3):
    M.append([])
    for j in xrange(i, 9, 3):
        M[-1].append(L[j])
 
print M
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

On peut faire ça en une seule ligne:

Code :

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

1
2
3
M = [[L[j] for j in xrange(i, 9, 3)] for i in xrange(0,3)]
print M
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Et même créer une fonction lambda si on a cette conversion à faire plusieurs fois:

Code :

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

1
2
3
4
creamat3x3 = lambda L: [[L[j] for j in xrange(i, 9, 3)] for i in xrange(0,3)]
M = creamat3x3(L)
print M
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

[fred1599]

Numpy est le module vraiment spécifique au travail de matrices, je le conseille fortement.

Voir exemple...

Bonne continuation...

[tyrtamos]

Et il y a encore plus simple:

Code :

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

1
2
3
4
L = [1,4,7,2,5,8,3,6,9]
 
print [[L[0], L[3], L[6]], [L[1], L[4], L[7]], [L[2], L[5], L[8]]]
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

[BufferBob]

et même encore plus simple avec une compréhension de liste et un slice

Code python :

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

1
2
3
4
>>> L = [1,4,7,2,5,8,3,6,9]
>>> P = [L[i::3] for i in xrange(len(L)/3)]
>>> P
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

[rosef]

merci beaucoup