파이썬 그리드 격자(meshgrid) 만드는 법과 원리

아래와 같은 네 점을 표현하고 싶다고 합시다. 

 

네 점의 좌표는 아래와 같습니다. 

 

(1,1) 
(1,2)
(2,1)
(2,2)

x와 y 따로 배열 형태로 입력해야합니다.  numpy array 를 이용하여 아래와 같이 입력할 수 있습니다. 

x=np.array([1,1,2,2])
y=np.array([1,2,1,2])

 

이정도는 할만 한데 아래와 같이 입력하고 싶은 경우를 생각해봅시다. 

 

25개의 점이므로 x 값 25개와 y값 25개를 각각 배열로 입력해야합니다. 이 과정을 편하게 해주는 함수가 meshgrid 입니다. 아래와 같이 입력하면 끝납니다. 

x=np.array([1,2,3,4,5])
y=np.array([1,2,3,4,5])
x,y=np.meshgrid(x,y)

x와 y는 각각 5by5 의 2차원 배열이 됩니다. x와 y를 중첩하면 같은 위치에 있는 점이 쌍을 이루어 좌표가 된다고 생각하시면 됩니다. x와 y를 출력해보면 아래와 같습니다. 

 

x
Out[37]: 
array([[1, 2, 3, 4, 5],
       [1, 2, 3, 4, 5],
       [1, 2, 3, 4, 5],
       [1, 2, 3, 4, 5],
       [1, 2, 3, 4, 5]])

y
Out[38]: 
array([[1, 1, 1, 1, 1],
       [2, 2, 2, 2, 2],
       [3, 3, 3, 3, 3],
       [4, 4, 4, 4, 4],
       [5, 5, 5, 5, 5]])