python之Pandas之将行中的值设置为自身乘以另一个值

假设我有下表:

multiply   total   tax 
Y          500     .1 
Y          250     .5 
N          300     .5 

只有当multiply 的值为Y 时，我才想将total 设置为total * tax

最终输出:

multiply   total   tax 
Y          50      .1 
Y          125     .5 
N          300     .5 

请您参考如下方法:

使用np.where

df['total'] = np.where(df['multiply'] == 'Y', df['total'] * df['tax'], df['total']) 
print(df) 
 
multiply  total  tax 
       Y   50.0  0.1 
       Y  125.0  0.5 
       N  300.0  0.5