python - 根据其他列修改数据框列中的值

Question

我有一个包含 A 列和 B 列的数据框 (df):

   A   B
0  'a' 50
1  'b' 25
2  'a' 30
3  'c' 200

如果 A 列的值为“a”，我想将 B 列的值乘以 100，所以我的输出应该是:

   A   B
0  'a' 5000
1  'b' 25
2  'a' 3000
3  'c' 200

任何人都可以建议一个更简洁的方法来做到这一点。谢谢。

Answer 1

我觉得测试子串没必要，只用DataFrame.loc比较值 a:

df.loc[df['A'] == "'a'", 'B'] *= 100
#same like
#df.loc[df['A'] == "'a'", 'B'] = df.loc[df['A'] == "'a'", 'B'] * 100
print (df)
   A     B
0  a  5000
1  b    25
2  a  3000
3  c   200