如何计算具有默认值的数据框的运行总和

Question

我将有两个数据框a和b，我希望a成为b默认值a's的运行总和。

>>> a = pd.DataFrame([[0, 0, 1], [0, 0, 0], [0, 0, 0]], columns=['a', 'b', 'c'])
>>> a
    a  b  c
 0  0  0  1
 1  0  0  0
 2  0  0  0
>>> b = pd.DataFrame([[1, 2, 3], [1, 2, 4], [4, 5, 2]], columns=['a', 'b', 'c'])
>>> b
   a  b  c
0  1  2  3
1  1  2  4
2  4  5  2

我想要的输出是：

    a  b  c
0   1  2  4
1   2  4  8
2   6  9  10

我试过a = b.cumsum()，但我得到了：

   a  b  c
0  1  2  3
1  2  4  7
2  6  9  9

Answer 1

您可以使用add将DataFrame与cumsum的总和值一起使用：

print (b.add(a).cumsum())
   a  b   c
0  1  2   4
1  2  4   8
2  6  9  10

详情：

print (b.add(a))
   a  b  c
0  1  2  4
1  1  2  4
2  4  5  2