python 新建一列_python – 如何在迭代pandas数据框时创建新列并插入行值

我正在尝试创建一个逐行迭代pandas数据帧的函数.我想基于其他列的行值创建一个新列.我的原始数据框可能如下所示：

df:

A B

0 1 2

1 3 4

2 2 2

现在我想在每个索引位置创建一个填充了列A – 列B的行值的新列,以便结果如下所示：

df:

A B A-B

0 1 2 -1

1 3 4 -1

2 2 2 0

我工作的解决方案,但只有当我不在函数中使用它时：

for index, row in df.iterrows():

print index

df['A-B']=df['A']-df['B']

这给了我想要的输出,但当我尝试将它用作函数时,我收到一个错误.

def test(x):

for index, row in df.iterrows():

print index

df['A-B']=df['A']-df['B']

return df

df.apply(test)

ValueError: cannot copy sequence with size 4 to array axis with dimension 3

我在这里做错了什么,我怎么能让它发挥作用？