python - 如何将填充应用于1？

Question

我有一个如下所示的数据框，

   A  B  C  D
0  1  0  0  0
1  0  1  0  0
2  0  1  0  0
3  0  0  1  0

我想把它转换成这样，

   A  B  C  D
0  1  0  0  0
1  1  1  0  0
2  1  1  0  0
3  1  1  1  0

到目前为止我已经尝试过，

df= df.replace('0',np.NaN)
df=df.fillna(method='ffill').fillna('0')

我的上面的代码工作正常，

但我认为还有其他更好的方法来解决这个问题，

Answer 1

使用cumsum将数据转换为数字，然后替换为 DataFrame.mask :

df = df.mask(df.astype(int).cumsum() >= 1, '1')
print (df)
   A  B  C  D
0  1  0  0  0
1  1  1  0  0
2  1  1  0  0
3  1  1  1  0

详细信息:

print (df.astype(int).cumsum())
   A  B  C  D
0  1  0  0  0
1  1  1  0  0
2  1  2  0  0
3  1  2  1  0

或者在numpy中使用numpy.where相同的原理:

arr = df.values.astype(int)

df = pd.DataFrame(np.where(np.cumsum(arr, axis=0) >= 1, '1', '0'), 
                  index=df.index, 
                  columns= df.columns)
print (df)
   A  B  C  D
0  1  0  0  0
1  1  1  0  0
2  1  1  0  0
3  1  1  1  0