Pandas从初始基数计算序列累计差值的无append优化方案求解
最优实现方案
你可以直接用pd.concat替代已经废弃的append方法,配合累计和逻辑就能实现需求,代码更简短易读:
total_series = start_value - pd.concat([pd.Series([0]), df['deduction']]).cumsum().reset_index(drop=True)
逻辑说明
- 在扣减序列的最前方拼接一个值为0的Series,计算该序列的累计和,得到的结果首位为0,后续每一位对应到当前位置的累计扣减总额
- 用初始基数直接减去这个累计和序列,最终得到的结果首位就是初始基数本身,后续每一位都是逐次扣减后的余额,和你原实现的输出完全一致。
如果追求更极致的性能,也可以用numpy实现:
import numpy as np total_series = pd.Series(start_value - np.cumsum(np.insert(df['deduction'].values, 0, 0)))
内容的提问来源于stack exchange,提问作者user17242583
相关产品推荐
相关产品推荐

