[시각화] heatmap

Ethan·2022년 5월 23일
0
  • 히트맵은 많은 양의 데이터와 현상을 수치에 따른 색상으로 표현한다.
    데이터 분포와 변수 간의 상관관계를 한 눈에 볼 수 있다는 장점이 있다.

  • 차원에 대한 제한은 없으나, 기본적으로 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")

profile
재미있게 살고 싶은 대학원생

0개의 댓글