小编jua*_*bin的帖子

我想将SAS代码传输到python，而在python中找不到保留函数。

数据就像：

type_id    amount
1           100
1           200
1           400
2           0
1           200
1           300
2           0
1           150

我想要的是type_id = 2时，金额等于上一行的负值。因此保留功能后的数据将如下所示：

type_id    amount
1           100
1           200
1           400
2          -400
1           200
1           300
2          -300
1           150

SAS代码为：

data B;
set A;
retain tempvar 0;
if type_id = 2
then amount = tempvar;
else tempvar = -amount;
drop tempvar;
run;

有谁知道如何在python中执行此操作吗？谢谢！