본문 바로가기
Python/Matplotlib

[Python] Matplotlib 관계 나타내는 차트 만들기(히트맵) plt.hist2d, sb.heatmap

by coding_su 2022. 11. 28.
  • 파이썬 히트맵 차트 만들기
>>> import numpy as np
>>> import pandas as pd
>>> import matplotlib.pyplot as plt
>>> import seaborn as sb

 

plt.hist2d() : 컬럼간의 관계(비례관계, 반비례관계, 아무관계없음)를 데이터가 많고 적음을 표시할 수 있도록 히트맵(Heat Maps)으로 시각화한 차트를 만들어준다

# hist2d 차트 만들기
>>> plt.hist2d(data= df, x= '컬럼1', y= '컬럼2')

>>> plt.hist2d(data= df, x= '컬럼1', y= '컬럼2', cmin= 0.5, cmap= 'viridis_r', bins= 20)
# cmin= 0.5 입력시 데이터 없는 부분은 표현되지 않는다
# cmap= 'viridis_r' 색을 반대로 표현하고 싶을때 입력
# bins= 20 구간 입력

>>> plt.colorbar()
# 입력시 컬러바 생성

sb.heatmap() : 컬럼을 히트맵(Heat Maps)으로 시각화한 차트를 만들어준다

>>> df = df[['컬럼1', '컬럼2', '컬럼3', '컬럼4', '컬럼5']]
>>> sb.heatmap(data= df, cmap= 'RdPu', annot= True, fmt='.1f', linewidths= 0.8)

# cmap= 색 변경
# annot= True 숫자 표시
# fmt='.1f 소숫점 몇자리 표시(정수는0)
# linewidths= 간격 지정

댓글