есть ли у кого-нибудь кусок кода для расчета скользящей кривой линейной регрессии на кадре данных pandas?
У меня уже есть функция для вычисления наклона для всего кадра данных, но я уверен, что какприменять / использовать его прокатки:
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()]
спасибо и ура, е.