Python Leetcode(867.转置矩阵)

Python Leetcode(867.转置矩阵)

给定一个矩阵 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]]

Solution:(新建一个空列表，遍历整个矩阵，将需要转换的值直接放在空列表对应的位置。)

class Solution(object):
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        result = []
        n_row = len(A)
        n_col = len(A[0])
        
        #这里需要注意的是，由于矩阵是一行一行添加的，所以转置时需要从原矩阵的列开始遍历，
        #转置过来才是行！
        for i in range(n_col):
            row = []
            for j in range(n_row):
                row.append(A[j][i])
            result.append(row)
        return result

solution = Solution()
print(solution.transpose([[1,2,3],[4,5,6],[7,8,9]]))

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