在Pandas中计算滚动回归并存储斜率 [英] Calculate a rolling regression in Pandas and store the slope

问题描述

我有一些时间序列数据，我想计算Pandas过去n天的GroupWise滚动回归，并将该回归的斜率存储在新列中。

我搜索了较旧的问题，它们要么没有得到回答，要么使用了Pandas OLS，我听说Pandas OLS已被弃用。

我想我可能可以将df.rolling.apply()与scipy.stats.linregress函数结合使用，但我找不出一个lambda函数来做我想做的事情。

以下是一些示例代码

import numpy as np
import pandas as pd
from scipy.stats import linregress

# make sample data
days = 21
groups = ['A', 'B', 'C']
data_days = list(range(days)) * len(groups)
values = np.random.rand(days*len(groups))

df = pd.DataFrame(data=zip(sorted(groups*days), data_days, values), 
                  columns=['group', 'day', 'value'])

# calculate slope of regression of last 7 days
days_back = 7

grouped_data = df.groupby('group')
for g, data in grouped_data:
    window = data.rolling(window=days_back,
                          min_periods=days_back)

我需要一个名为‘SLOPE’的新列，从第7天开始，在该列中存储过去7天的线性回归的斜率。

推荐答案

我有一些错误的假设，首先我不需要遍历组，其次我并不真正了解rolling.apply是如何工作的...

下面是(看起来)可以工作的代码。我使用了scipy.stats中的linregress函数：

import numpy as np
import pandas as pd
from scipy.stats import linregress

# create random sample data
days = 14
groups = ['A', 'B', 'C']
data_days = list(range(days)) * len(groups)
values = np.random.rand(days*len(groups))

df = pd.DataFrame(data=zip(sorted(groups*days), data_days, values), 
                  columns=['group', 'day', 'value'])

def get_slope(array):
    y = np.array(array)
    x = np.arange(len(y))
    slope, intercept, r_value, p_value, std_err = linregress(x,y)
    return slope


# calculate slope of regression of last 7 days
days_back = 3

df['rolling_slope'] = df.groupby('group')['value'].rolling(window=days_back,
                               min_periods=days_back).apply(get_slope, raw=False).reset_index(0, drop=True)

print(df)

这篇关于在Pandas中计算滚动回归并存储斜率的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！