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

Question

假设我有下表:

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

Answer 1

使用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