Python pandas nonzero cumsum

Question

我想在pandas的python的数据框中应用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