Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Matplotlib / Seaborn 플롯의 특정 셀에 사용자 정의 테두리 추가

<시간/>

Matplotlib/Seaborn 플롯의 특정 셀에 사용자 정의 테두리를 추가하려면

단계

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • 일부 열이 있는 데이터 프레임을 만듭니다.
  • 행렬 데이터세트를 계층적으로 클러스터된 히트맵으로 플로팅합니다.
  • 하위 플롯 배열로 히트맵 축을 가져옵니다.
  • Matplotlib의 특정 셀에 사용자 정의 테두리를 추가하기 위해 border_color. 변수를 초기화할 수 있습니다.
  • 맞춤 테두리 색상을 사용하여 히트맵 축에 직사각형 패치를 추가합니다.
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import pandas as pd
from matplotlib import pyplot as plt
import seaborn as sns
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
df = pd.DataFrame({"col1": [1, 4, 2, 3, 5], "col2": [3, 4, 1, 5, 2]})
g = sns.clustermap(df, figsize=(7.50, 3.50))
ax = g.ax_heatmap
border_color = "yellow"
ax.add_patch(plt.Rectangle((1, 2), 2, 1, fill=False,
edgecolor=border_color, lw=5))
plt.show()

출력

Matplotlib / Seaborn 플롯의 특정 셀에 사용자 정의 테두리 추가