Heat Map

히트 맵이란?

히트 맵(heat map)은 열을 뜻하는 히트(heat)와 지도를 뜻하는 맵(map)을 결합시킨 단어로, 색상으로 표현할 수 있는 다양한 정보를 일정한 이미지 위에 열분포 형태의 비주얼한 그래픽으로 출력하는 것이 특징이다.

# 필요한 라이브러리를 불러옵니다.
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

# 깃허브에 준비된 데이터를 가져옵니다.
!git clone https://github.com/taehojo/data.git

# 피마 인디언 당뇨병 데이터셋을 불러옵니다.
df = pd.read_csv('./data/pima-indians-diabetes3.csv')

colormap = plt.cm.gist_heat    # 그래프의 색상 구성을 정합니다.
plt.figure(figsize=(12,12))    # 그래프의 크기를 정합니다.

sns.heatmap(df.corr(), linewidths=0.1, vmax=0.5, cmap=colormap, 
linecolor='white', annot=True)
plt.show()
# heatmap 함수를 사용해서 출력하기

위 그림에서 가장 눈여겨보아야 할 부분은 당뇨병 발병 여부를 가리키는 ➊ diabetes 항목입니다. diabetes 항목을 보면 pregnant부터 age까지 상관도가 숫자로 표시되어 있고, 숫자가 높을수록 밝은 색상으로 채워져 있습니다.

diabetes 열을 보면 diabetes를 뺀 나머지 중에서 가장 밝은 칸은 plasma입니다. 따라서 diabetes와 plasma의 수치 관계만을 따로 분리해서 아래 그림과 같이 그래프나 히스토그램으로 나타낼 수 있습니다.