主页 / 问题 / 882069
Accepted
Viktorov
Asked:2020-09-17 04:53:26 +0000 UTC2020-09-17 04:53:26 +0000 UTC

如何在 pandas 中构建浮动分组窗口？

  • 772

我有一个数据集：

id  |  value
------------
1   |  3
2   |  4
3   |  6
4   |  7
5   |  2

我想在此处添加一列，其中包含前 3 行的平均值，同时考虑到当前行，因此结果是这样的：

id  | value | my_column
-----------------------
1   | 3     |  3                   3/1
2   | 4     |  3.5             (3+4)/2
3   | 6     |  4.33          (3+4+6)/3
4   | 7     |  5.66          (4+6+7)/3
5   | 2     |  5             (6+7+2)/3

到目前为止，我已经编写了一个函数，可以多次循环我的集合并计算这些总和。似乎可以不用循环。能？

python

1 个回答

  1. Best Answer
    In[114]: df
Out[114]: 
   id  value
0   1      3
1   2      4
2   3      6
3   4      7
4   5      2

    您可以使用大小为每列s的滑动窗口：3min_periods=1df.value

    In[115]: df["my_column"] = df.value.rolling(window=3, min_periods=1).mean()
In[116]: df
Out[116]: 
   id  value  my_column
0   1      3   3.000000
1   2      4   3.500000
2   3      6   4.333333
3   4      7   5.666667
4   5      2   5.000000
    • 2

相关问题