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

Seaborn 히트맵에서 각 타일에 테두리(프레임) 넣는 방법

Seaborn의 히트맵에서 각 타일 주위에 테두리(프레임)를 그리려면 heatmap() 메서드에서 linewidths(선 두께)와 linecolor(선 색상) 옵션을 사용하면 됩니다. 이 두 파라미터를 조절하면 셀 사이의 경계를 원하는 색과 굵기로 명확하게 표현할 수 있습니다.

구현 단계

  • 그림 크기를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
  • Pandas 데이터프레임을 생성합니다. 이 예제에서는 5개의 열을 가진 데이터를 사용합니다.
  • heatmap() 메서드를 호출하여 직사각형 데이터를 색상으로 인코딩된 행렬 형태로 시각화합니다. 이때 linewidthslinecolor 값을 지정해 타일 테두리를 만듭니다.
  • 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((5, 5)), columns=["col1", "col2", "col3", "col4", "col5"])

sns.heatmap(df, linewidths=4, linecolor='green')

plt.show()

실행 결과

Seaborn 히트맵에서 각 타일에 테두리(프레임) 넣는 방법

위 코드를 실행하면 초록색(green) 테두리가 적용된 히트맵이 생성됩니다. linewidths 값을 키우면 테두리가 더 두꺼워지고, linecolor를 다른 색상으로 변경하면 원하는 스타일의 프레임을 손쉽게 구현할 수 있습니다.