试图在python中将矩阵从3x4转置为4x3

Question

我必须从命令行读取文件，并设置矩阵行和列。当我运行我的程序时（如下）：

python3 transpose.py               matrix                                                 3 4
        (current code below) (file with numbers 1 2 3 4 5 6 7 8 9 10 11 12 13)  (matrix row,cols)

一起打印应该返回：

The empty matrix is:[[0,0,0,0],[0,0,0,0],[0,0,0,0]]
The original matrix is:
1 2 3 4
5 6 7 8
9 10 11 12

The transposed matrix is:
1 5 9
2 6 10
3 7 11
4 8 12

这是我的 transpose.py 我花了大约 10 个小时在它上面，但无法让它工作，我只是想不出还能做什么！

 import sys
from scanner import *


def main():
    readinput(sys.argv[1],[2])
    size = 3
    rows = size
    cols = rows
    makeList(size)
    matrix(rows,cols)
    manipulateMatrix(matrix(rows,cols))
    print(manipulateMatrix(matrix(rows,cols)))
    print("The original matrix is: ")
    displayMatrix(manipulateMatrix(matrix(rows,cols)))
    transposeSquare(manipulateMatrix(matrix(rows,cols)),size)
    print (transposeSquare(manipulateMatrix(matrix(rows,cols)),size))
    print("The transposed matrix: ")
    displayMatrix(transposeSquare(manipulateMatrix(matrix(rows,cols)),size))    
def readinput(filename,matrix):
    s = Scanner(filename)
    r = s.readtoken()
    while r != "":
        r = int(r)
        c = s.readint()
        v = s.readint()
        matrix[r][c]=v
        r = s.readtoken()
    s.close()

def makeList(size):
    lyst = []
    for i in range(size):
        lyst = lyst + [None]
    return lyst

def matrix(rows,cols):
    matrix = makeList(rows)
    for i in range(rows):
        matrix[i] = makeList(cols)
        return matrix

def manipulateMatrix(m):
    rows = len(m)
    cols = len(m[0])
    count = 1
    for r in range(0,rows,1):
        for c in range(0,cols,1):
            m[r][c] = count
            count += 1
    return m

def transposeSquare(m,size):
    for r in range(0,size):
         for c in range(0,size):
            m[r][c],m[c][r] = m[c][r],m[r][c]
    return m

def displayMatrix(m):
    m == rows == cols
    for r in range(0,size):
        for c in range(0,cols,1):
            print(m[r][c],end = "")
        print()
    return          
main()

现在我被困在

 Traceback (most recent call last):
  File "transpose.py", line 66, in <module>
    main()
  File "transpose.py", line 6, in main
    readinput(sys.argv[1],[2])
  File "transpose.py", line 27, in readinput
    matrix[r][c]=v
IndexError: list index out of range

Answer 1

转置 2D 列表要简单得多。将zip() function 与* 解包运算符一起使用：

>>> matrix = [[1,2,3,4], [5,6,7,8], [9,10,11,12]]
>>> new = list(zip(*matrix))  # call to list not necessary in Python 2
>>> new
[(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)]

或者，如果你不想要元组：

>>> new = [list(tup) for tup in zip(*matrix)]
>>> new
[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

Answer 2

使用 numpy 库。它是为此目的而设计的，在 C/C++ 代码中运行数字，并且比在 python 中运行并且必须调试它要快得多。

import numpy as np
m = np.array(<your list matrix here>)
print m.shape # to see the shape of your matrix
mt = m.T #m.T transpose the matrix
print m.shape
print m