Python Pandas 非零 cumsum

Question

我想在 python 中的 pandas 数据帧上应用 cumsum，但没有零。只是我想保留零并在数据帧上执行 cumsum 。假设我有这样的数据框:

import pandas as pd

df =  pd.DataFrame({'a' : [1,2,0,1], 
                    'b' : [2,5,0,0], 
                    'c' : [0,1,2,5]})

   a  b  c
0  1  2  0
1  2  5  1
2  0  0  2
3  1  0  5

结果应该是

   a  b  c
0  1  2  0
1  3  7  1
2  0  0  3
3  4  0  8

有什么想法可以避免循环吗？在 R 中有 ave 函数，但我对 python 很陌生，我不知道。

Answer 1

您可以屏蔽 df，以便只覆盖非零单元格:

In [173]:
df[df!=0] = df.cumsum()
df

Out[173]:
   a  b  c
0  1  2  0
1  3  7  1
2  0  0  3
3  4  0  8