基于新客户号码的累积和重置

Question

我有一个像这样的数据集。

​

我想做的是：

• 每一个客户号码，金额信贷金额，金额借方金额(最老至最新)，直到借方累计=信贷金额*-1
• 信用证可适用于部分发票。
• 返回一份文件编号(和金额)列表，以便为每个客户应用信贷。

df['debcum_sum'] = df.groupby(['CustNum'])['Debits'].apply(lambda x: x.cumsum())

我本来打算为累积和添加一个列，但我想尝试并将其作为一个for循环。有什么建议吗？

Answer 1

使用groupby + cumsum的组合，然后使用pd.Series.where/mask根据Credit列隐藏值-

v = df.groupby(['CustNum'])['Debits'].cumsum()
df['debit_cumsum'] = v.where(v >= df['Credit'] * -1)