상관계수 code

상관계수 코드

 

import seaborn as sns

 

# 상관관계 matrix 작성

correlation_matrix = train.corr()

 

# 보통 heatmap으로 표현!!
mask = np.triu(np.ones_like(correlation_matrix, dtype = bool))
# 하삼각을 가리려면 np.tril

plt.figure(figsize = (15, 15))
sns.heatmap(correlation_matrix, annot = True, cmap = "coolwarm", fmt = ".2f", mask = mask, vmin = -1, vmax = 1)
plt.title("correlation matrix")
plt.show()

 

# 상관계수 값 그래프에 표시

# heatmap 색 지정

# 상관계수 값 소수점 정리

# 전체 볼 필요 없이 반만 나오도록!

# 최대, 최소값 고정 (그래야 시각화해서 보기 편함)