熊猫股票回归图表

Question

我想创建一个简单的线性回归图表，就像在excel中一样。以最短的方式。

使用pandas .p​​lot绘制带有回归线的股票收益图表的最简单方法是什么？

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)