python - 行是 3 个最新值的总和

Question

df = pd.DataFrame({'values':[2,6,1,5,8,9]})

我正在尝试创建一个名为 sum 的新 rom，它是 3 个最新值的总和。所以它会返回这个 [np.nan, np.nan, 9, 12, 14, 22]

Answer 1

使用rolling与 Rolling.sum :

print (df.rolling(3).sum())
   values
0     NaN
1     NaN
2     9.0
3    12.0
4    14.0
5    22.0

或者:

print (df['values'].rolling(3).sum())
0     NaN
1     NaN
2     9.0
3    12.0
4    14.0
5    22.0
Name: values, dtype: float64

df['sum'] = df['values'].rolling(3).sum()
print (df)
   values   sum
0       2   NaN
1       6   NaN
2       1   9.0
3       5  12.0
4       8  14.0
5       9  22.0

Docs .