본문 바로가기
Python/Matplotlib

[Python] Matplotlib 상관계수 구하기 corr()

by coding_su 2022. 12. 2.
  • 파이썬 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()

댓글