python - 如何将数据帧的两列传递给Python中的函数？

Question

我正在尝试下面的代码。但我返回错误

("Newmethod() missing 1 required positional argument: 'B'", 'occurred at index 0')

def Newmethod(A,B):
  Add=A+B
  return Add

Df=pd.DataFrame([[1,2,3],[4,5,6],[7,8,9]],columns=["A","B","C"])
Df['D']=Df[['A','B']].apply(Newmethod, axis=1)

我在这里想念什么？请帮助我。

Answer 1

改变它

Df['D']=Df[['A','B']].apply(lambda x:Newmethod(x.A,x.B), axis=1)

   A  B  C   D
0  1  2  3   3
1  4  5  6   9
2  7  8  9  15

解释

您可以使用np.sum或更改您的Newmethod()。否则，您只能在表单中执行此操作。

import numpy as np
Df['D']=Df[['A','B']].apply(np.sum, axis=1)
# or
def Newmethod(x):
    Add= x.A + x.B
    return Add
Df['D']=Df[['A','B']].apply(Newmethod, axis=1)