[python] 연결된 배열 이동하기

N×N인 격자에서 1번 행은 N번과 연결되어 있고, 1번 열은 N번 열과 연결되어 있는 경우의 구현

dx = [-1, 0, 1, 0]
dy = [0, -1, 0, 1]

def move(x, y, s, d):
    # d방향으로 s만큼 이동시키기
    nx, ny = (x+dx[d]*s)%N, (y+dy[d]*s)%N
    return nx, ny