题目:
给定一个矩阵 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
分析:
- 定义一个全为0的矩阵,然后将原矩阵的第i行第j列的元素赋为新矩阵的第j行第i列的值
代码:
class Solution:
def transpose(self, A: List[List[int]]) -> List[List[int]]:
row = len(A)
col = len(A[0])
matrix = [[0 for i in range(row)] for j in range(col)]
for i in range(row):
for j in range(col):
matrix[j][i] = A[i][j]
return matrix
结果: