[Python] 2차원 행렬을 90도 회전시키는 방법 ☑️

n x n 의 2차원 행렬을 90도(시계방향) 회전시키는 방법

 

def rotate90(matrix):  # 회전시키는 행렬은 nxn로 가정
	n = len(matrix)
	newMatrix = [[0 for _ in range(n)] for _ in range(n)]
	for y in range(n):
		for x in range(n-1, -1, -1):
			newMatrix[y][n-1-x] = matrix[x][y]
	return newMatrix

 

 

import sys

n, m = map(int,sys.stdin.readline().split())

print("[Original matrix]")
matrix = [[row*m+col for col in range(m)] for row in range(n)]

for i in range(n):
    for j in range(m):
        print(matrix[i][j], end=" ")
    print()
print()


print("[Rotate Right 90]")
matrix_r90= [[0 for col in range(n)] for row in range(m)]

for i in range(n):
    for j in range(m):
        matrix_r90[j][n-i-1] = matrix[i][j]

for i in range(m):
    for j in range(n):
        print(matrix_r90[i][j], end=" ")
    print()
print()


print("[Rotate Right 90]")
matrix_l90= [[0 for col in range(n)] for row in range(m)]

for i in range(n):
    for j in range(m):
        matrix_r90[m-j-1][i] = matrix[i][j]

for i in range(m):
    for j in range(n):
        print(matrix_r90[i][j], end=" ")
    print()
print()