PYTHON 정리(히트맵 색상 변경)(24.01.12)

matplotlib 라이브러리를 사용해서 히트맵으로 상관계수를 표현하고자할때

히트맵의 색깔을 바꾸고싶을 수도 있다.

 

히트맵의 색깔을 바꾸고싶다면

히트맵 코드를 작성할 때

plt.pcolor(변수명, cmap=컬러맵 선택)

을 추가해주면 된다.

 

예를들어

plt.figure(figsize=(14,5))

plt.pcolor(corr4)

plt.xticks(np.arange(0.5, len(corr4.columns), 1), corr4.columns, rotation=45)

plt.yticks(np.arange(0.5, len(corr4.index), 1), corr4.index)

plt.title('Outlet 상관관계', fontsize=20, pad=20)

plt.pcolor(corr4, cmap='OrRd')

plt.colorbar()

plt.show()

위와 같이 코드를 입력했다면 아래와 같은 히트맵을 확인할 수 있다.

cmap 뒤에 입력하는 컬러맵은 아래 링크에 다양하게 소개되어있으니 참고하면 되겠다.

참고로 예시에서 사용한 컬러맵은 'OrRd'이다.

https://matplotlib.org/3.3.1/tutorials/colors/colormaps.html

Choosing Colormaps in Matplotlib — Matplotlib 3.3.1 documentation