熊猫按滚动总和分组

Question

我的数据如下所示

id  var window
1   0   3
1   0   3
1   9   3
1   1   3
1   0   3
1   10  5
1   0   12
2   0   12
2   3   4
2   0   12
2   5   12

我想做的是根据“ var”的值计算“ id”的滚动总和。如果“ var”不为零，则计算滚动总和，滚动窗口的长度取自“ window”列；如果“ var”为零，则滚动总和等于0。

下面是预期的结果

   id   var window  roll_sum
    1   0   3       0
    1   0   3       0
    1   9   3       9
    1   1   3       10
    1   0   3       0
    1   10  5       20
    1   0   12      0
    2   0   12      0
    2   3   4       3
    2   0   12      0 
    2   5   12      8

有人可以帮忙吗？