Transponuj macierz w Pythonie

Question

Próbuję utworzyć funkcję transpozycji macierzy w języku Python. Macierz to dwuwymiarowa tablica, reprezentowana jako lista list całkowitych. Na przykład, po to macierz 2X3 (czyli wysokość matrycy 2 i szerokości 3)

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

Do transpozycji elementu j w indeksie i-tego należy się element ego w jth index. Oto jak wygląda powyższa próbka:

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

Jak mogę to zrobić?

Answer 1

Można użyć zip z * dostać transpozycję macierzy:

>>> 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)] 

Jeśli chcesz, aby zwrócona lista była listą:

>>> [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

Gdybyśmy chcieli wrócić tą samą matrycę chcielibyśmy napisać:

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

Co to robi to iteracje nad matrycę M przechodząc przez każdego wiersza i powrocie każdy element w każdej kolumnie. więc kolejność będzie jak:

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

teraz na pytanie 3, zamiast tego chcą przejść przez kolumnę kolumny, wracając każdy element w każdym wierszu. więc kolejność będzie jak:

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

Dlatego właśnie zmienić kolejność, w której iterację:

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

Answer 3

Czy istnieje nagroda za bycie leniwym i korzystanie z funkcji transpozycji macierzy NumPy? ;)

import numpy as np 

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

b = a.transpose()