Python 线性回归曲线

有没有人有一段代码来计算熊猫数据帧上的滚动线性回归曲线

---

我已经有了一个计算整个数据帧斜率的函数，但我知道如何应用/使用它：

def slope_intercept(x_val, y_val):
    x = np.array(x_val)
    y = np.array(y_val)
    m = ( ( (np.mean(x)*np.mean(y) ) - np.mean(x*y)) /
        ( ( np.mean(x)*np.mean(x)) - np.mean(x*x)))

    m = round(m,2)
    b=(np.mean(y)-np.mean(x)*m)
    b=round(b,2)
    return m,b

m,b=slope_intercept(df1.index.tolist(), df1['Close'].tolist())
df1['reg_line'] = [(m*x)+b for x in df1.index.tolist()]

谢谢大家，干杯，

---

e、

“我已经有了一个函数来计算整个数据帧的斜率”-你现在怎么做？@AmiTavory我已经将代码扩展到任何人之上了？坦斯克！