Seaborn 히트맵에서 사용자 정의 그리드 라인 만드는 방법

Seaborn 히트맵에서 사용자 정의 그리드 라인(격자선)을 만들려면 heatmap() 메서드의 linewidths(선 두께)와 linecolor(선 색상) 매개변수를 활용하면 됩니다. 이 두 옵션을 조절하면 셀 사이의 경계를 원하는 스타일로 표현할 수 있습니다.

구현 단계

• 그림 크기를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.

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

• heatmap() 메서드를 사용해 데이터를 색상으로 인코딩된 행렬 형태로 시각화합니다. 이때 linewidths와 linecolor 값을 지정합니다.

• 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()

실행 결과

위 코드를 실행하면 각 셀 사이에 두께 4의 초록색 그리드 라인이 적용된 히트맵이 생성됩니다. linewidths 값을 늘리면 격자선이 더 굵어지고, linecolor를 다른 색상으로 변경하면 원하는 색의 구분선을 얻을 수 있습니다.