Transpose a matrix in Python

Question

Sto provando a creare una funzione di trasposizione di matrici in Python. Una matrice è una matrice bidimensionale, rappresentata come una lista di liste di numeri interi. Ad esempio, la seguente è una matrice 2X3 (cioè l'altezza della matrice è 2 e la larghezza è 3):

A=[[1, 2, 3], 
    [4, 5, 6]] 

essere recepita la voce jth nell'indice esimo dovrebbe diventare l'elemento i-esimo del jth index. Ecco come appare il campione sopra riportato:

>>> transpose([[1, 2, 3], 
       [4, 5, 6]]) 
[[1, 4], 
[2, 5], 
[3, 6]] 
>>> transpose([[1, 2], 
       [3, 4]]) 
[[1, 3], 
[2, 4]] 

Come posso fare questo?

Answer 1

È possibile utilizzare zip con * per ottenere la trasposizione di una matrice:

>>> A = [[ 1, 2, 3],[ 4, 5, 6]] 
>>> zip(*A) 
[(1, 4), (2, 5), (3, 6)] 
>>> lis = [[1,2,3], 
... [4,5,6], 
... [7,8,9]] 
>>> zip(*lis) 
[(1, 4, 7), (2, 5, 8), (3, 6, 9)] 

Se si desidera che l'elenco restituito sia un elenco di elenchi:

>>> [list(x) for x in zip(*lis)] 
[[1, 4, 7], [2, 5, 8], [3, 6, 9]] 
#or 
>>> map(list, zip(*lis)) 
[[1, 4, 7], [2, 5, 8], [3, 6, 9]] 

Answer 2

Volendo restituire la stessa matrice avremmo scrivere:

return [[ m[row][col] for col in range(0,width) ] for row in range(0,height) ] 

Quello che fa è che itera su una matrice m passando attraverso ogni riga e ritornare ogni elemento in ciascuna colonna. Così l'ordine sarebbe come:

[[1,2,3], 
[4,5,6], 
[7,8,9]] 

Ora per la domanda 3, che invece vogliono andare colonna per colonna, tornando ogni elemento in ogni riga. Così l'ordine sarebbe come:

[[1,4,7], 
[2,5,8], 
[3,6,9]] 

Pertanto basta passare l'ordine in cui iteriamo:

return [[ m[row][col] for row in range(0,height) ] for col in range(0,width) ] 

Answer 3

Esiste un premio per essere pigri e utilizzare la funzione di trasposizione degli array NumPy? ;)

import numpy as np 

a = np.array([(1,2,3), (4,5,6)]) 

b = a.transpose()