python - 将 cumsum 限制为仅前 4 行

Question

初学者的问题：

我想在我的数据框上创建一个累积总和列，但我只希望该列添加前 4 行（包括当前行）的值。我还需要对框架中的每个新“类型”重新开始计数。

这就是我想要的：

Type    Value    Desired column
  A       1        -
  A       2        -
  A       1        -
  A       1        5
  A       2        6
  A       2        6
  B       2        -
  B       2        -
  B       2        -
  B       2        8
  B       1        7
  B       1        6

Answer 1

您可以通过在我们的类型后面加上applya来做到这一点。例如：rolling_sumgroupby

>>> df["sum4"] = df.groupby("Type")["Value"].apply(lambda x: pd.rolling_sum(x,4))
>>> df
   Type  Value  sum4
0     A      1   NaN
1     A      2   NaN
2     A      1   NaN
3     A      1     5
4     A      2     6
5     A      2     6
6     B      2   NaN
7     B      2   NaN
8     B      2   NaN
9     B      2     8
10    B      1     7
11    B      1     6

pandasNaN用来表示缺失数据；如果你真的想要-，你也可以这样做，使用

df["sum4"] = df["sum4"].fillna('-')