[파이썬][Numpy] 2D 및 3D 그래프 그리기 예제

24 Aug 2023

numpy

Matplotlib를 사용하여 2D 및 3D 그래프를 그리는 예제를 제공하겠습니다. 2D 그래프는 일반적으로 선 그래프, 막대 그래프, 히스토그램 등과 같은 2차원 데이터를 시각화하는 데 사용되며, 3D 그래프는 3차원 데이터를 시각화하는 데 사용됩니다.

2D 그래프 예제:

import numpy as np
import matplotlib.pyplot as plt

# 데이터 생성
x = np.linspace(0, 10, 100)
y = np.sin(x)

# 2D 선 그래프 그리기
plt.figure(figsize=(8, 6))
plt.plot(x, y, label='sin(x)', color='blue', linestyle='-', linewidth=2)
plt.title('2D 선 그래프')
plt.xlabel('X 축')
plt.ylabel('Y 축')
plt.legend()
plt.grid(True)
plt.show()

위의 코드는 이전 2D 그래프 예제와 동일합니다.

3D 그래프 예제:

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

# 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))

# 3D 표면 그래프 그리기
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection='3d')
surface = ax.plot_surface(X, Y, Z, cmap='viridis')
fig.colorbar(surface, shrink=0.5, aspect=5)
ax.set_title('3D 표면 그래프')
plt.show()

위의 코드에서는 다음과 같은 작업을 수행합니다:

NumPy를 사용하여 3D 데이터를 생성합니다. 이 예제에서는 3D 사인 함수를 생성합니다.
Matplotlib에서 mpl_toolkits.mplot3d를 임포트하여 3D 플롯을 생성할 수 있도록 합니다.
3D 표면 그래프를 생성하고 설정합니다. 색상 맵을 추가하고 축 레이블을 설정합니다.

실행 결과로 3D 표면 그래프가 표시됩니다.

Matplotlib를 사용하면 2D 및 3D 데이터를 시각화하고 다양한 그래프 유형을 생성할 수 있습니다. 원하는 형태의 데이터를 시각적으로 이해하기 위해 Matplotlib를 활용할 수 있습니다.