transpose_matrix.py (757B)
| # 867. Transpose Matrix
"""
given a 2d integer array 'matrix', return the transpose of 'matrix'. the
transpose of a matrix is the matrix flipped over its main diagonal switching
the matrix's rows and column indices
"""
class Solution(object):
def transpose(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: List[List[int]]
"""
n = len(matrix)
m = len(matrix[0])
ans = [[0] * n for _ in range(m)]
for i in range(n):
for j in range(m):
ans[j][i] = matrix[i][j]
return ans
if __name__ == "__main__":
obj = Solution()
print(obj.transpose(matrix=[[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
print(obj.transpose(matrix=[[1, 2, 3], [4, 5, 6]]))
|
