[파이썬] 등고선 그래프 그리기
등고선 그래프는 두 개의 변수에 대한 2D 그래프를 표현하는 강력한 시각화 도구입니다. 이 그래프는 등고선으로 이루어진 등고선 플롯이라는 방식으로 그려집니다. 파이썬에서는 matplotlib
라이브러리를 사용하여 간단하게 등고선 그래프를 그릴 수 있습니다.
Matplotlib 설치하기
먼저, matplotlib
라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 pip
명령어로 설치할 수 있습니다.
pip install matplotlib
등고선 그래프 그리기 예제
다음은 간단한 예제 코드를 통해 파이썬으로 등고선 그래프를 그리는 방법을 설명합니다.
import matplotlib.pyplot as plt
import numpy as np
# 등고선 그래프를 그릴 함수
def plot_contour():
# x, y 범위 설정
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
# x, y 좌표 생성
X, Y = np.meshgrid(x, y)
# 등고선 값을 계산
Z = np.sin(np.sqrt(X**2 + Y**2))
# 등고선 그래프 그리기
plt.contour(X, Y, Z, levels=10, colors='k')
# 그래프 보여주기
plt.show()
# 등고선 그래프 그리기 함수 호출
plot_contour()
위 예제는 matplotlib.pyplot
모듈을 plt
로 임포트하고, numpy
모듈을 np
로 임포트하는 것으로 시작됩니다. 예제에서는 -5
부터 5
까지의 범위에서 x
와 y
를 생성하고, meshgrid
함수를 사용하여 X
와 Y
좌표 그리드를 생성합니다.
그런 다음, 등고선 값을 계산하기 위해 Z
를 np.sin
함수를 사용하여 계산합니다. 마지막으로 plt.contour
함수를 사용하여 등고선 그래프를 그립니다. levels
파라미터를 사용하여 등고선 레벨의 수를 조절할 수 있으며, colors
파라미터를 사용하여 등고선의 색상을 지정할 수 있습니다. plt.show
를 호출하여 그래프를 표시합니다.
이제 위의 예제 코드를 실행하면 파이썬에서 등고선 그래프를 생성할 수 있습니다.
위 예제 코드를 참고하여 원하는 데이터에 대해 등고선 그래프를 그려보세요!