Python

Python 데이터프레임을 sb.heatmap 으로 나타내기

yugyeong 2022. 11. 30. 17:56

 

아래에 0부터 100 사이의 수치로 되어있는 데이터프레임이 있다.

이 데이터프레임의 수치 비교를 쉽게 하기 위해서 데이터프레임 자체를 heatmap 으로 나타내려한다.

 

 

데이터를 쉽게 분석하기 위해서 데이터가 많으면 진한 색으로 나타내고, 데이터가 적으면 연한 색으로 나타나게 할 것이다.  

 

 

import matplotlib.pyplot as plt
import seaborn as sb

필요한 라이브러리를 임포트 해준다.

 

 

sb.heatmap() 안에 data 파라미터에는 데이터프레임명( df2 )을 넣어준다.

아래와같이 데이터프레임 자체가 heatmap 으로 표현되었다.

현재 아래의 데이터프레임은 데이터가 많으면 연한색, 데이터가 적으면 진한색으로 나타나게 설정되어있다.

 

이제부터 몇몇 파라미터를 이용하여 수치 데이터가 차트에 나타나게 하고, 데이터가 많을 수록 진한 색을 나타나게 할 것이다.

 

 

 

먼저, cmap 파라미터를 cmap= RdPu 로 할 경우 데이터가 많을수록 진한색을 나타나고 데이터가 적을수록 연한 색을 나타나게 한다.

annot= True 로 하면 heatmap 에 데이터프레임의 수치 데이터가 아래처럼 표현된다.

fmt 는 format을 의미한다. 아래의 예시에서 작성한 ' .1f ' 는 소숫점 아래로 한자리 수까지만 표현한다는 의미이다.

linewidths 는 데이터프레임 행,열의 너비를 의미한다.

 

마지막으로 plt.figure 는 차트의 너비와 높이를 의미한다. 보기 편하게 너비와 높이를 튜플 형식으로 작성하여 주면 된다.