Python Grid 시각화

코딩테스트를 볼 때 grid를 이용해 풀어야 하는 문제가 나올 때가 있습니다.

grid가 머릿속에 잘 그려지면 굳이 시각화를 해서 눈으로 보면서 풀지 않아도 되지만 grid가 머릿속에 잘 안그려진다면 시각화를 하면서 문제에 접근해보면 실마리를 찾을 수 있을 것입니다.

import matplotlib.pyplot as plt

size = 10
grid = [[0] * size for _ in range(size)]

for x in range(size):
    for y in range(size):
        if x == y:
            grid[y][x] = 1

plt.imshow(grid)
plt.gca().invert_yaxis()
plt.show()

위 코드를 실행하면 우상향 하는 그래프와 같이 array의 모양을 시각화 해서 볼 수 있습니다.

import matplotlib.pyplot as plt

def get_grid(x_size, y_size):
    grid = [[0] * x_size for _ in range(y_size)]
    return grid

x_size = 5
y_size = 10
grid = get_grid(x_size, y_size)

for x in range(x_size):
    for y in range(y_size):
        if x == y:
            grid[y][x] = 1

plt.imshow(grid)
plt.gca().invert_yaxis()
plt.show()

위 코드는 x, y size를 입력받아 해당 사이즈의 grid를 return하게 수정한 코드 입니다.
행열로는 10행 5열입니다.

end.