본문 바로가기
Coding/Python

heatmap 그래프 사용법

by Tony.B 2020. 4. 8.

- 모듈 설치

pip install seaborn

- 기본 문법

import seaborn as sns
import matplotlib.pyplot as plt

cols_view = ['a','b','c','d']
sns.set(font_scale=1.5)
hm = sns.heatmap(A,                       # 여기서 A 값은 numpy 배열
                cbar = True,              # heatmap 색에 대한 기준바(?)를 표시할지 말지를 결정
                annot  = True,            # haetmap 위에 A 값을 표시 할지 말지 결정
                square = True,            # heatmap을 정사각형으로 만들지 말지 결정
                fmt = '.2f',              # values 값의 소수점 표기
                annot_kws = {'size':15},  # heatmap 위의 A값의 글씨 크기
                yticklabels = cols_view,  # y축 이름 설정
                xticklabels = cols_view)  # x축 이름 설정

plt.tight_layout()
plt.show()

위 코드의 결과 화면