python - 在 df.apply() 中的自定义函数中传递 Pandas DataFrame 中的不同列

Question

假设我有一个数据框df:

  x y z
0 1 2 3
1 4 5 6
2 7 8 9

我想要两个新列，分别是 x * y 和 x * z:

  x y z xy xz
0 1 2 3  2  3
1 4 5 6 20 24
2 7 8 9 56 63

因此我定义了一个函数 func(仅作为示例)，它将字符串 'y' 或字符串 'z' 作为一个参数，用于指示我要将哪一列与列 x 相乘:

def func(row, colName):
    return row['x'] * row[colName]

并将该函数应用于数据框df:

df['xz'] = df.apply(func, axis=1)

显然这里是错误的，因为我没有指定 colName、'y' 或 'z'。问题是，df.apply() 只接受函数名，我如何告诉它接受两个参数？

Answer 1

我认为 eval 在这里很完美

df['x*y'],df['x*z']=df.eval('x*y'),df.eval('x*z')
df
Out[14]: 
   x  y  z  x*y  x*z
0  1  2  3    2    3
1  4  5  6   20   24
2  7  8  9   56   63