Matplotlib은 파이썬에서 데이터를 시각화하기 위한 강력한 도구입니다. 2D 그래프 뿐만 아니라 3D 그래프를 그릴 수도 있습니다. 이 포스트에서는 Matplotlib을 사용하여 3D 그래프를 그리는 방법에 대해 알아보겠습니다.

1. 3D 그래프를 그리기 위한 모듈 가져오기

우선, 3D 그래프를 그리기 위해 mpl_toolkits.mplot3d 모듈을 가져와야 합니다. 이 모듈을 사용하여 3D 그래프를 생성할 수 있습니다.

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

2. 3D 그래프 그리기

이제 3D 그래프를 그려보겠습니다. 먼저 3D 그래프를 그릴 데이터를 만들어야 합니다. 여기서는 간단한 예제로 3D 함수로부터 데이터를 생성하여 그래프를 그리겠습니다.

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
x, y = np.meshgrid(x, y)
z = np.sin(np.sqrt(x**2 + y**2))

ax.plot_surface(x, y, z, cmap='viridis')

plt.show()

위 코드에서는 np.meshgrid를 사용하여 2D 그리드를 생성하고, np.sin을 사용하여 3D 함수를 계산했습니다. 그리고 ax.plot_surface를 사용하여 3D 플롯을 만들었습니다.

3. 그래프 커스터마이징

그래프를 더욱 멋지게 만들기 위해 다양한 옵션을 적용할 수 있습니다. 예를 들어, 그래프의 제목, 축 레이블, 색상 등을 추가하거나 변경할 수 있습니다.

ax.set_title('3D Surface Plot')
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.view_init(45, 240)

위 코드에서 ax.set_title, ax.set_xlabel, ax.set_ylabel, ax.set_zlabel을 사용하여 각각 그래프의 제목과 축 레이블을 설정하고, ax.view_init을 사용하여 그래프의 시야각을 조정했습니다.

4. 결론

이제 Matplotlib을 사용하여 간단한 3D 그래프를 그리는 방법에 대해 알아보았습니다. Matplotlib을 사용하면 데이터를 시각적으로 표현하는 다양한 방법을 쉽게 구현할 수 있습니다.

더 많은 정보를 원하시면 Matplotlib 공식 문서를 참고하시기 바랍니다.