Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Matplotlib에서 등고선 플롯의 색상 막대에 대한 제한 설정

<시간/>

Matplotlib에서 countour 플롯의 색상 막대에 대한 제한을 설정하려면 다음 단계를 수행할 수 있습니다. -

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • x 만들기 및 y numpy를 사용한 데이터 포인트
  • x를 사용하여 데이터 가져오기 및 y .
  • 좌표 벡터에서 좌표 행렬을 가져옵니다.
  • vmin 초기화 및 vmax matplotlib에서 등고선 플롯의 색상 막대에 대한 제한을 설정합니다.
  • contourf()를 사용하여 등고선 그리기 방법.
  • vmin 범위 내에서 매핑 가능한 스칼라를 사용하여 컬러바 만들기 및 vmax .
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

from matplotlib import pyplot as plt
import numpy as np
from matplotlib.cm import ScalarMappable

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.arange(20)
y = np.arange(20)

data = x[:, None] + y[None, :]
X, Y = np.meshgrid(x, y)

vmin = 0
vmax = 15

fig, ax = plt.subplots()

qcs = ax.contourf(
      X, Y, data,
      vmin=vmin, vmax=vmax
)

fig.colorbar(
   ScalarMappable(norm=qcs.norm, cmap=qcs.cmap),
   ticks=range(vmin, vmax+5, 5)
)

plt.show()

출력

Matplotlib에서 등고선 플롯의 색상 막대에 대한 제한 설정