python - 如何获得条件累计和

Question

我有一列需要条件累积和，其中空值重置累积。我如何实现这一目标？

t = pd.DataFrame([2, 3, None, 1, 4])

# DataFrame
| 0    |
|------|
| 2    |
| 3    |
| NaN  |
| 1    |
| 4    |

# Desired Output
| 0    |
|------|
| 2    |
| 5    |
| NaN  |
| 1    |
| 5    |

标签： pythonpandas

Answer 1

像这样：

In [29]: t[0].groupby(t[0].isna().cumsum()).cumsum()                                                                                                                                                        
Out[29]: 
0    2.0
1    5.0
2    NaN
3    1.0
4    5.0
Name: 0, dtype: float64