파이썬 2차원 배열 쉽게 회전 시키기

알고리즘 풀다보면 상하좌우 여러방향 혹은 열단위로 작업을 수행해야 하는 경우가 있다.

 

이럴 때 경우에 따라 작업하기 편한 한 방향(주로 직관적이며 슬라이싱을 이용하기 편한 좌 -> 우로 한다)에 대해서만 코드 를 작성하고 나머지 방향은 배열을 회전 시켜서 적용하는 방법이 있다.

 

이 경우 코드가 훨씬 짧아지고 깔끔해지며, 에러가 일어날 일이 적다는 장점이 있지만 반대로 시간 복잡도가 늘어나고 경우에 따라 예외 케이스로 인해 오답이 출력되는 경우도 있으니 조심해서 적용해야 할 것이다.

 

원래를 이중 for문을 도는 방법이 정석(?)이지만 실제 문제 풀 때 쓰기 편하라고 각 방향 별로 한 줄씩 함수 형태로 작성했다.

 

만약 학습이 목적이라면 for문으로 각 방향별로 + 전치배열까지 직접 구현해보는 것 추천!!

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

# 시계 방향으로 회전
def rotate(d):
    # 0도
    if d == 0:
        return arr
    # 90도 
    elif d == 1:
        return list(map(list, zip(*arr[::-1])))
    # 180도
    elif d == 2:
        return [[arr[N - r - 1][N - c - 1] for c in range(N)] for r in range(N)]
    # 270도
    else:
        return list(map(list, zip(*arr)))[::-1]

 

아래는 이해를 돕기 위한 출력 예시이다.

for d in range(4):
    print(d)
    for row in rotate(d):
        print(*row)
    print()
    
'''
0
1 2 3
4 5 6
7 8 9

1
7 4 1
8 5 2
9 6 3

2
9 8 7
6 5 4
3 2 1

3
3 6 9
2 5 8
1 4 7
'''