每次出现 NaN 时如何创建运行总计并重新启动它？

Question

我想在每次遇到 nan 时启动一个新的运行总计 例如，从所附图片中，它将前 3 个值求和 [1242536, 379759, 1622295]，然后显示运行总计 3244590.0，然后它将从第 5 个值开始新的运行总计，直到第 9 个，显示这些值的总和等等。我想将这些运行总计放在这些 NaN 值旁边的新列中。

我已尝试通过以下方式解决此问题：

for i in df['Budget_Expenditure_2012_']:
        if np.isnan(i) == True:
            x = pd.Index(df['Budget_Expenditure_2012_']).get_loc(i)

    
print(x)

for item in range(0, len(x) - 1, 2):
    second_list.append([x[item],x[item + 1]])    
print(second_list)

这个想法是找到每对行之间的值的总和。这对将是需要求和的每个范围的开始位置和最后位置。 在这一点上，我迷失了如何执行这个求和运算。

Answer 1

使用shift、isna 和cumsum 的组合到gropuby，然后transform，最后分配列为nan 的结果值

df.loc[df['Budget_Expenditure_2012_'].isna(), 'new_column'] = (
    df.groupby(
        df.Budget_Expenditure_2012_.shift()
                                   .isna()
                                   .cumsum()
    )['Budget_Expenditure_2012_'].transform('sum')
)

Answer 2

使用此代码，您可以在名为“Totals”的新列上获取每个 nan 的“运行总计”。

total = 0
df['Totals'] = 0 # assign 0 initially to all rows of the new column

for i in range(df.shape[0]): # shape[0] return number of rows

    expenditure = df.loc[i+1, 'Budget_Expenditure_2012_'] # i+1 coz your indexing starts at 1

    if np.isnan(expenditure):
        df.loc[i, 'Totals'] = total
        total = 0
    else:
        total += expenditure