2

早上好，

我有以下数据框：

a = [1,2,3,4,5,6]
b = pd.DataFrame({'a': a})

我想创建一个列，对列“a”的下一个“n”行求和，包括 a 的当前值；我试过：

n = 2
b["r"] = pd.rolling_sum(b.a, n) + a 
print(b)
   a     r
0  1   NaN
1  2   5.0
2  3   8.0
3  4  11.0
4  5  14.0
5  6  17.0

拥有以下内容将是令人愉快的：

   a     r
0  1   1 + 2 + 3 = 6
1  2   2 + 3 + 4 = 9
2  3   3 + 4 + 5 = 12
3  4   4 + 5 + 6 = 15
4  5   5 + 6 + 0 = 11
5  6   6 + 0 + 0 = 6
4

1 回答 1

2

熊猫 >= 1.1

Pandas 现在支持“前瞻性窗口操作”，请参见此处。

从 1.1 开始，您可以使用FixedForwardWindowIndexer

idx = pd.api.indexers.FixedForwardWindowIndexer
b['a'].rolling(window=idx(window_size=3), min_periods=1).sum()

0     6.0
1     9.0
2    12.0
3    15.0
4    11.0
5     6.0
Name: a, dtype: float64

请注意，这对于日期时间滚动操作仍然（在撰写本文时）非常有问题 - 请谨慎使用。

熊猫 <= 1.0.X

如果没有内置支持，您可以通过首先反转数据、使用rolling_sumwithmin_periods=1并再次反转来获得输出。

b.a[::-1].rolling(3, min_periods=1).sum()[::-1]

0     6.0
1     9.0
2    12.0
3    15.0
4    11.0
5     6.0
Name: a, dtype: float64
于 2018-04-16T10:06:42.273 回答