python-3.x - lambda pandas 数据框中的行减法

Question

我有一个包含多列的数据框。其中一栏是累计收入栏。如果这一年没有结束，那么剩余时间里的收入将保持不变，因为接下来的每日收入为 0。

数据框看起来像这样

现在我想创建一个新列，其中该行被最后一行减去，如果结果为 0，则在新列中为该行打印 0。如果不为零，则使用行值。新的数据框应如下所示:

我的想法是使用 apply lambda 方法来做到这一点。所以这就是想法:

{df['2017new'] = df['2017'].apply(lambda x: 0 if row - lastrow == 0 else x)}

但我不知道如何编写代码的 row-lastrow 部分。这个怎么做？提前致谢!

Answer 1

通过使用np.where

df2['New']=np.where(df2['2017'].diff().eq(0),0,df2['2017'])
df2
Out[190]: 
   2016  2017  New
0    10    21   21
1    15    34   34
2    70    40   40
3    90    53   53
4    93    53    0
5    99    53    0