상관관계를 확인하는 그래프 - Heatmap & Scatter 그래프

feature 간 상관관계를 보여주는 그래프의 두가지 타입 : heatmap 그래프와 Scatter(산점도) 그래프

 

1. Heatmap 그래프 그려보기

👉 모든 컬럼간 상관관계를 점으로 보여주는 것. (색이 짙을시 상관관계가 높음)

doc.corr() #doc라는 데이터프레임의 feature 간 상관관계를 보여줌.
doc2 = doc.corr() #doc2라는 변수에 해당 데이터셋을 저장

doc2.iplot(kind='heatmap') #iplot으로 히트맵 그래프 그리기

옵션 : colorscale = 'ylorrd' 등이 있음. 

 

2. Scatter 그래프 그려보기

👉 두 컬럼간 상관관계를 점으로 보여주는 것. (우상향시 상관관계가 높음)

doc.iplot(kind='scatter', x = 'A', y = 'B', mode = 'markers')
# x와 y값에 비교할 컬럼을 지정

fig = go.Figure()
fig.add_trace(
	go.Scatter(
    	x = doc['A'], y = ['B'], mode = 'markers'
        )
    )
fig.show()