基于另一列的值对一列Pandas DF进行条件运算

Question

我在下面有两列pandas数据框。

df = pd.DataFrame({'quantity':[15, 30, 35, 20], 'sign':[1,1,2,1]})

我想应用条件语句并将每行中的quantity值乘以-1，如果该行中的相应sign值等于2.我不想创建新列。 我只想将其应用于quantity列。

我尝试过使用.loc和.apply(lamda x:) ，但我不能正确应用它们。 任何帮助表示赞赏。

Answer 1

使用DataFrame.loc与面具由Series.eq由标量平等和多：

df = pd.DataFrame({'quantity':[15, 30, 35, 20], 'sign':[1,1,2,1]})

df.loc[df['sign'].eq(2), 'quantity'] *=  - 1
print (df)
   quantity  sign
0        15     1
1        30     1
2       -35     2
3        20     1