히트맵은 많은 양의 데이터와 현상을 수치에 따른 색상으로 표현한다.
데이터 분포와 변수 간의 상관관계를 한 눈에 볼 수 있다는 장점이 있다.
차원에 대한 제한은 없으나, 기본적으로 2차원 그래프로 표현한다.
일부 데이터는 pd.DataFrame.pivot
기능을 활용해 변형해 주어야 한다.
import pandas as pd
import seaborn as sns
data = sns.load_dataset("flights")
flights = pd.DataFrame(data)
# 피보팅
pivot = flights.pivot(index='year', columns='month', values='passengers')
# heatmap
sns.heatmap(pivot)
# annot 옵션: 수치 표시
sns.heatmap(pivot, linewidths=.2, annot=True, fmt="d")
# 색상 변경
sns.heatmap(pivot, cmap="YlGnBu")