- 파이썬 Matplotlib 상관계수 구하기
상관계수는 -1부터 1까지의 값을 갖는다
1일 때 완벽한 양의 상관관계가 되고, -1일 때 완벽한 음의 상관관계가 된다
0이라면 별다른 상관관계가 없음을 의미한다
.corr() : 상관계수 구하기
# 데이터 프레임의 전체 상관계수 구하기
>>> df.corr()
# 두 컬럼간의 상관계수 구하기
>>> df[['컬럼1','컬럼2']].corr()
구한 상관계수를 보기 편하게 히트맵으로 만들기
df_corr = df[['컬럼1', '컬럼2', '컬럼3', '컬럼4']].corr()
sb.heatmap(data= df_corr, annot=True, fmt='.1f', cmap= 'coolwarm', linewidths= 0.7, vmin= -1, vmax= 1)
plt.show()
plt.figure(figsize=(20,10)) # 차트 사이즈 조절
sb.heatmap(data= df_all, annot=True, fmt='.1f', cmap= 'coolwarm', linewidths= 0.5, vmin= -1, vmax= 1)
plt.show()
댓글