Matplotlib에서 등고선 플롯을 만드는 방법

등고선 플롯은 등고선을 사용하여 3차원 데이터를 2차원으로 시각화할 수 있는 플롯 유형입니다.

다음 두 함수를 사용하여 Matplotlib에서 등고선 플롯을 생성할 수 있습니다.

• matplotlib.pyplot.contour() – 등고선 플롯을 생성합니다.
• matplotlib.pyplot.contourf() – 채워진 등고선 플롯을 생성합니다.

다음 예에서는 이 두 기능을 실제로 사용하는 방법을 보여줍니다.

예 1: Matplotlib의 등고선 플롯

Python에 다음과 같은 데이터가 있다고 가정합니다.

 import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

다음 코드를 사용하여 데이터에 대한 등고선 플롯을 만들 수 있습니다.

 import matplotlib.pyplot as plt

plt. outline (X, Y, Z, colors=' black ')

플롯에 한 가지 색상만 사용하는 경우 점선은 음수 값을 나타내고 실선은 양수 값을 나타냅니다.

대안은 cmap 인수를 사용하여 컬러맵을 지정하는 것입니다. 또한 level 인수를 사용하여 플롯에 사용할 더 많은 선을 지정할 수도 있습니다.

 plt. contour (X, Y, Z, levels= 30 , cmap=' Reds ') 

우리는 ‘Reds’ cmap을 사용하기로 선택했지만 Matplotlib 문서 페이지 에서 색상 팔레트 옵션의 전체 목록을 찾을 수 있습니다.

예제 2: Matplotlib의 채워진 등고선 플롯

채워진 윤곽 경로는 선 사이의 공간이 채워진다는 점을 제외하면 윤곽 경로와 유사합니다.

다음 코드는 Contourf() 함수를 사용하여 이전 예제에서 사용된 것과 동일한 데이터에 대해 채워진 등고선 플롯을 생성하는 방법을 보여줍니다.

 plt. contourf (X, Y, Z, cmap=' Reds ') 

colorbar() 함수를 사용하여 플롯 옆에 레이블이 지정된 색상 막대를 추가할 수도 있습니다.

 plt. contourf (X, Y, Z, cmap=' Reds ')
plt. colorbar () 

여기에서 더 많은 Matplotlib 튜토리얼을 찾을 수 있습니다.