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

Python에서 녹색에서 빨간색까지의 히트 맵을 만드는 방법은 무엇입니까? (매트플롯립)

<시간/>

녹색에서 빨간색 범위의 히트맵을 Python에서 생성하려면 다음 단계를 수행할 수 있습니다. -

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • 다양한 색상의 사전을 만드세요.
  • LinearSegmentedColormap을 사용하여 선형 매핑 세그먼트에서 컬러맵 생성 .
  • 그림과 서브플롯 세트를 생성합니다.
  • 5☓5 차원의 임의 데이터 생성
  • 비정규 직사각형 그리드로 의사 색상 플롯을 만듭니다.
  • ScalarMappable용 컬러바 생성 인스턴스, *매핑 가능* .
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import matplotlib.pyplot as plt
import matplotlib.colors as colors
import numpy as np

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

cdict = {'red': ((0.0, 0.0, 0.0),
                (0.5, 1.0, 1.0),
                (1.0, 1.0, 0.7)),

        'green': ((0.0, 0.7, 0.7),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0)),

         'blue': ((0.0, 0.0, 0.0),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0))
         }
GnRd = colors.LinearSegmentedColormap('GnRd', cdict)

fig, ax = plt.subplots(1)

data = np.random.rand(5, 5)*6.-3.

p = ax.pcolormesh(data, cmap=GnRd, vmin=-5, vmax=5)

fig.colorbar(p, ax=ax)

plt.show()

출력

Python에서 녹색에서 빨간색까지의 히트 맵을 만드는 방법은 무엇입니까? (매트플롯립)