python - 对 pandas 数据框的列执行累积求和，忽略 NAN

Question

我有一个 pandas 数据框，如下所示。我想根据每个订单对“NEW1”列执行累积和。下面的代码可以部分工作，但它不会忽略 Nan 的意思我预计最后一行的“cumsum”值为 8

import pandas as pd
import numpy as np
df = pd.DataFrame({'ORDER':["A", "A", "B", "B"], 'NEW1':[np.nan, 5, 8, np.nan]})
df['cumsum'] = df.groupby(['ORDER'])['NEW1'].cumsum()
df

    ORDER   NEW1    cumsum
0   A       NaN     NaN
1   A       5.0     5.0
2   B       8.0     8.0
3   B       NaN     NaN

我的预期输出:

    ORDER   NEW1    cumsum
0   A       NaN     NaN
1   A       5.0     5.0
2   B       8.0     8.0
3   B       NaN     8.0

Answer 1

fillna() 在 groupby 之前，并使用 transform

df['cumsum']=df.fillna(0).groupby('ORDER')['NEW1'].transform('cumsum')



ORDER  NEW1  cumsum
0     A   NaN     0.0
1     A   5.0     5.0
2     B   8.0     8.0
3     B   NaN     8.0