python - 将函数应用于以其他列作为参数的 pandas 列

Question

嘿，我正在尝试使用y作为参数将函数应用于列x，因此对于每一行我想应用x给y，像这样:

def fun(x, y):
    return x + y



df['xy'] = df['x'].apply(fun, args = df['y'])

但是它不起作用。有任何想法吗？请注意，我真正的功能并不像添加两个值那么简单

Answer 1

使用DataFrame.apply使用 lambda 函数和用于每行处理的 axis=1 :

df['xy'] = df.apply(lambda x: fun(x['x'], x['y']), axis=1)

但如果可能的话，更好的是使用矢量化操作，在这里:

df['xy'] = df['x'] + df['y']