Python - 基于行值的乘法

Question

我有一个数据框和三个变量 x、y、z。

x = 10, y = 20, z = 30

df = pd.DataFrame({'A':['a','b','c'],
                   'B':[6,7,8]})

这是我想做的:

创建一个新列“C”:

If df['A'] == 'a', df['C'] = df['B']*x
If df['A'] == 'b', df['C'] = df['B']*y
If df['A'] == 'c', df['C'] = df['B']*z

有什么快速的方法吗？

Answer 1

创建一个字典，将列 A 值映射到乘法值，然后使用 map在 A 列上乘以 B 列:

mul_map = {'a': 10, 'b': 20, 'c': 30}
df['C'] = df['B'] * df['A'].map(mul_map)

结果输出:

   A  B    C
0  a  6   60
1  b  7  140
2  c  8  240