스터디하다, 생각보다 heatmap mask를 계단식으로 표현하는 법을 모르는 사람들이 많아 글을 써본다

나도 문법까먹었을때 보면 되겠다

heat map mask설정하기

1.mask에 사용할 mask를 생성한다

여기서는 계단모양으로 설정하려 한다

아래코드는 대각선을 기준으로 좌측 아래는 계단모양으로 0을 그 이외에는 1인 matrix를 생성한다

np.triu(np.ones_like(corr, dtype=np.bool))

(출처-https://medium.com/@szabo.bibor/how-to-create-a-seaborn-correlation-heatmap-in-python-834c0686b88e)

2. heatmap에서 mask를 지정한다

heatmap = sns.heatmap(df.corr(), mask=mask, vmin=-1, vmax=1, annot=True)

실행시 위 matrix 모양으로 히트맵이 생성된다

---

참고

• https://medium.com/@szabo.bibor/how-to-create-a-seaborn-correlation-heatmap-in-python-834c0686b88e