python - Pandas - 将滚动应用于列速度

Question

我有一个数据框，其中我只取数字列的子集，计算每个数字列的 5 天滚动平均值并将其作为新列添加到 df。

这种方法可行，但目前需要相当长的时间(每列 8 秒)。我想知道是否有更好的方法来做到这一点。

我目前正在做的工作玩具示例:

data = {'Group': ['A','A','A','A','A','A','B','B','B','B','B','C','C','C','C','C'],
        'Year' : ['2017', '2017', '2017', '2018', '2018', '2018', '2017', '2017', '2018', '2018', '2017', '2017', '2017', '2017', '2018', '2018'],
        'Score 1' : [1,2,3,4,5,6,1,2,3,4,5,1,2,3,4,5],
       'Score 2': [1,4,5,7,5,5,6,1,4,5,6,7,4,6,4,6]}
df = pd.DataFrame(data)

for col in ['Score 1', 'Score 2']:
    df[col + '_avg'] = df.groupby(['Year', 'Group'])[col].apply(lambda x: x.rolling(2, 1).mean().shift().bfill())

Answer 1

对于任何登陆此的人，我能够通过先排序并避免使用 lambda 函数来显着加快速度:

return_df[col + '_avg'] = df.sort_values(['Group', 'Year']).groupby(['Group'])[col].rolling(2,1).mean().shift().values