[파이썬][Numpy] np.meshgrid 그리드 형태 배열 생성 방법

np.meshgrid() 함수를 사용하면 그리드 형태의 배열을 생성할 수 있습니다. 이 함수는 입력으로 x와 y 좌표를 받아 그리드 형태의 좌표 매트릭스를 반환합니다. 이를 통해 2D 그래프를 그리거나 함수를 평가하는 데 유용합니다. 아래는 np.meshgrid() 함수를 사용하여 그리드 형태의 배열을 생성하는 예제입니다.

import numpy as np
import matplotlib.pyplot as plt

# x와 y 좌표 벡터 생성
x = np.linspace(-5, 5, 10)  # -5부터 5까지 10개의 점
y = np.linspace(-5, 5, 10)

# 그리드 생성
X, Y = np.meshgrid(x, y)

# X와 Y는 그리드 형태의 2D 배열
print("X 배열:")
print(X)
print("Y 배열:")
print(Y)

# 그리드 포인트 시각화
plt.scatter(X, Y)
plt.xlabel('X 좌표')
plt.ylabel('Y 좌표')
plt.title('그리드 포인트')
plt.show()

이 코드에서 np.linspace() 함수를 사용하여 x와 y 좌표를 생성하고, 이를 np.meshgrid() 함수에 전달하여 그리드 형태의 X와 Y 배열을 생성합니다. 그리드 포인트를 시각화하기 위해 matplotlib을 사용합니다.

np.meshgrid() 함수는 2D 그리드를 생성할 때 주로 사용되지만, 3D 그리드나 다차원 그리드도 생성할 수 있습니다. 이러한 그리드를 사용하여 함수의 값을 여러 점에서 계산하거나 2D 또는 3D 플롯을 그릴 수 있습니다.