팬더 주식 회귀 차트

Question

나는 Excel에서와 같이 간단한 선형 회귀 차트를 만들고 싶습니다. 가능한 한 최단 경로로.

pandas .plot을 사용하여 회귀선이있는 주식 수익률 차트를 플로팅하는 가장 쉬운 방법은 무엇입니까?

Answer 1

그것이 있어야하지 statsmodels

import statsmodels.api as sm 
mod = sm.OLS.from_formula('y ~ x', data=df) # y and x are column names in the DataFrame 
res = mod.fit() 
fig, ax = plt.subplots() 
sm.graphics.abline_plot(model_results=res, ax=ax) 
df.plot(kind='scatter', x='x', y='y', ax=ax)