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

Matplotlib에서 히트맵의 위/아래 삼각형만 플로팅

<시간/>

matplotlib에서 히트맵의 위/아래 삼각형만 플롯하려면 numpy를 사용하여 마스크된 2D 배열을 가져오고 이를 이미지로 변환하여 히트맵을 생성할 수 있습니다.

단계

  • Figure 크기를 설정하고 서브플롯 사이와 주변의 패딩을 조정합니다.

  • 5×5 차원의 임의 데이터를 생성합니다.

  • numpy.tri() 사용 주어진 대각선 아래에 1이 있고 다른 곳에 0이 있는 배열을 만드는 방법입니다.

  • 마스크된 배열을 사용하여 마스크된 2D 배열 데이터를 가져옵니다(3단계 사용).

  • imshow() 사용 데이터를 2D 일반 래스터와 같이 이미지로 표시하는 방법입니다.

  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(5, 5)
mask = np.tri(data.shape[0], k=-1)
data = np.ma.array(data, mask=mask)
plt.imshow(data, interpolation="nearest", cmap='copper')
plt.show()

출력

Matplotlib에서 히트맵의 위/아래 삼각형만 플로팅