[Python] 2차원 리스트 90도 회전

90도 회전하면 행과 열이 반대된다는 것만 기억하기!

🎨 [구현 코드]

def rotate_90_degree(matrix):
  n = len(matrix) # 행 길이
  m = len(matrix[0]) # 열 길이
  result = [[0] * n for _ in range(m)] # rotate 한 결과는 행과 열이 반대됨
  for i in range(n):
    for j in range(m):
      result[j][n - 1 - i] = matrix[i][j]
  return result


result = rotate_90_degree([[1, 2], 
                           [3, 4], 
                           [5, 6]])
for i in result:
  print(i)

🎨 [실행 결과]

[5, 3, 1]
[6, 4, 2]