计算期间的Pandas动态索引参考

时间:2014-07-16 21:19:39

标签: dynamic reference pandas

我有以下数据框

    val  sum
0    1    0
1    2    0
2    3    0
3    4    0
4    5    0
5    6    0
6    7    0

我想计算下三行的总和(包括当前行)值。我需要为非常大的文件执行此操作。什么是最有效的方式?预期的结果是

    val  sum
0    1    6
1    2    9
2    3    12
3    4    15
4    5    18
5    6    13
6    7    7

一般情况下,如何在进行作业时动态引用其他行(通过布尔运算)?

1 个答案:

答案 0 :(得分:1)

> pd.rolling_sum(df['val'], window=3).shift(-2)
0     6
1     9
2    12
3    15
4    18
5   NaN
6   NaN

如果您想要将最后的值填入"填写"那么你需要在你的数据帧结束时使用NaN&#39。