[python+plotly] px.scatter 플롯에 추세선 넣기

plotly를 이용해서 산점도를 그릴 때 추세선(trendline)을 넣고 싶은 경우가 있다. 추세선을 넣어주면 데이터의 장기적인 증감 추세를 좀 더 한 눈에 볼 수 있기 때문이다. 추세선을 넣는 방법은 매우 간단하다. 

 

px.scatter(df, x="total_bill", y="tip", trendline="ols") 

 

그런데 여기서 에러가 나는 분들이 있을 것이다. 아마도 statsmodels 모듈과 관련된 에러일 것이다. 이 경우에는 statsmodels를 설치해주면 된다.

 

pip install statsmodels

 

만약 추세선의 색상을 별도로 지정하고 싶을 때는 trendline_color_override를 설정해주면 된다.

 

px.scatter(df, x="total_bill", y="tip", trendline="ols", trendline_color_override="red") 

 

 

 

참고자료

[1] https://towardsdatascience.com/scatter-plots-with-plotly-express-1b7f5579919b