상관행렬그래프를 깔끔하게 반만 그릴 때
•
np.triu : 상삼각행렬 만들기
np.triu(array, k=0)
◦
array: 2차원의 행렬 형태를 입력
◦
k = 0(default) : 행렬의 가장 첫 행에서 0의 시작 개수(일반적으로 0 아님 1)
•
np.ones_like : 입력된 array의 동일한 shape 으로 모든 행렬의 원소가 1인 행렬 생성
np.ones_like(matrix/vector, dtype)
◦
matrix/vector : 행렬형태거나 vector 형태로 입력
◦
dtype은 np.float32, np.bool, np.int8, np.int32
corr = df[연속형 변수들의 칼럼 리스트].corr().abs()
mask = np.triu(np.ones_like(corr, dtype=np.bool)) # 나머지 원소가 모두 1인 상삼각행렬 만들기(corr과 동일한 행렬 shape)
fig,ax = plt.subplots(figsize=(14,14))
# plot heatmap
sns.heatmap(corr, mask=mask, annot=True, fmt='.2f', cmap='coolwarm',
cbar_kws={'shrink':.8}, vmin=0, vmax=1) # vmin, vmax -> 색상 농도 지정
# ytick
plt.yticks(rotation=0)
plt.show()
Python
복사