1. 程式人生 > >leetcode-867. 轉置矩陣[python]

leetcode-867. 轉置矩陣[python]

給定一個矩陣 A, 返回 A 的轉置矩陣。 矩陣的轉置是指將矩陣的主對角線翻轉,交換矩陣的行索引與列索引。

示例 1: 輸入:[[1,2,3],[4,5,6],[7,8,9]] 輸出:[[1,4,7],[2,5,8],[3,6,9]]

示例 2: 輸入:[[1,2,3],[4,5,6]] 輸出:[[1,4],[2,5],[3,6]]

提示: 1 <= A.length <= 1000 1 <= A[0].length <= 1000

class Solution(object):
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        B = [[0 for i in range(len(A))] for i in range(len(A[0]))]
        for i in range(len(A)):
            for j in range(len(A[i])):
                B[j][i] = A[i][j]
                
        return B         

注意for迴圈的範圍。