데이터 분석 공부/Python
PYTHON 정리(히트맵 색상 변경)(24.01.12)
세조목
2024. 1. 12. 17: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
Note Click here to download the full example code Choosing Colormaps in Matplotlib Matplotlib has a number of built-in colormaps accessible via matplotlib.cm.get_cmap. There are also external libraries like [palettable] and [colorcet] that have many extra
matplotlib.org