Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Seaborn 히트맵에서 색상 막대(Colorbar) 숨기는 방법

Seaborn의 heatmap() 메서드로 히트맵을 그릴 때, 색상 막대(colorbar)가 함께 표시되는 것이 기본 동작입니다. 만약 시각화 결과에서 색상 막대를 제거하고 싶다면 cbar=False 옵션을 사용하면 됩니다. 이 옵션 하나만 추가하면 간단하게 색상 막대를 숨길 수 있습니다.

구현 단계

  • 그림(figure) 크기를 설정하고, 서브플롯 사이와 주변의 여백(padding)을 자동으로 조정합니다.

  • 4개의 열을 가진 데이터프레임(DataFrame)을 생성합니다.

  • heatmap() 메서드를 사용해 데이터를 색상으로 인코딩된 행렬 형태의 히트맵으로 시각화하며, 이때 cbar=False를 지정하여 색상 막대를 숨깁니다.

  • show() 메서드를 호출하여 최종 그래프를 화면에 출력합니다.

예제 코드

import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame(np.random.random((4, 4)), columns=["a", "b", "c", "d"])
sns.heatmap(df, cbar=False)

plt.show()

실행 결과

위 코드를 실행하면 4x4 크기의 무작위 데이터로 구성된 히트맵이 생성되며, 오른쪽에 표시되던 색상 막대 없이 히트맵만 깔끔하게 나타납니다.

Seaborn 히트맵에서 색상 막대(Colorbar) 숨기는 방법

추가 팁

반대로 색상 막대를 유지하면서 위치나 크기를 조정하고 싶다면 cbar_kws 파라미터를 활용할 수 있습니다. 예를 들어 sns.heatmap(df, cbar_kws={'shrink': 0.8})처럼 작성하면 색상 막대의 크기를 조절할 수 있어, 다양한 상황에 맞게 히트맵을 커스터마이징할 수 있습니다.