[Python] 배열 회전

우회전

def rotate(arr):
    arr_len = len(arr)
    save_arr = [ [0] * arr_len for _ in range(arr_len) ]

    for c in range(arr_len):
        for r in range(arr_len):
            save_arr[r][arr_len - 1 - c] = arr[c][r]
    return save_arr


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

input_arr = rotate(input_arr)
print(input_arr)

# print
# [[5, 9, 5, 1], [4, 8, 6, 2], [3, 7, 7, 3], [2, 6, 8, 4]]