根据条件更改列的符号

Question

输入DF：

value1, value2
123L, 20
222S, 10
222L, 18

我想在值value2中的值中取值，其中value1中的L字母为负，因此我想将它们乘以-1

扩展结果：

value1, value2
    123L, -20
    222S, 10
    222L, -18

我的密码

if np.where(DF['value1'].str.contains('L', case=False)):
    DF['value2'] = DF['value2'] * -1

但是在输出中，我正在接收列value2所有值都是负数。 如何仅对选定的行实施此条件？ 谢谢

Answer 1

您可以对loc使用布尔索引：

df.loc[df['value1'].str[-1] == 'L', 'value2'] *= -1

或者，使用pd.Series.mask ：

df['value2'].mask(df['value1'].str[-1] == 'L', -df['value2'], inplace=True)

如果您热衷于使用np.where ，则可以这样np.where ，但很冗长：

df['value2'] = np.where(df['value1'].str[-1] == 'L', -df['value2'], df['value2'])

注意np.where已经被矢量化了，您不应该将它与if一起使用。

Answer 2

str.endswith + loc

df.loc[[x.endswith('L') for x in df.value1], 'value2'] *= -1
df

  value1  value2
0   123L     -20
1   222S      10
2   222L     -18

---

mask

df['value2'] = df.value2.mask(df.value1.str.endswith('L'), -df.value2)
df

  value1  value2
0   123L     -20
1   222S      10
2   222L     -18

Answer 3

从hack replace 更多信息

df.value1.replace({'L':-1,'S':1},regex=True)*df.value2.astype(int)
Out[183]: 
0   -20
1    10
2   -18
dtype: int64

-

分配回去

df.value2*=df.value1.replace({'L':-1,'S':1},regex=True)
df
Out[187]: 
  value1  value2
0   123L     -20
1   222S      10
2   222L     -18