кривая линейной регрессии Python Pandas - PullRequest
0 голосов
/ 18 мая 2018

есть ли у кого-нибудь кусок кода для расчета скользящей кривой линейной регрессии на кадре данных 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()]

спасибо и ура, е.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...