给定一个看起来像这样的数据框
Vote A Vote B
1 4
3 2
1 5
我想添加一个名为 Winner 的新列,用于比较 Vote A 和 Vote B 两列之间的值并指定获胜者。
Winner
B
A
B
我如何在 Pandas 中执行此操作?
请您参考如下方法:
你可以使用
def winner(row):
if row['Vote A'] > row['Vote B']:
return 'A'
elif row['Vote A'] < row['Vote B']:
return 'B'
else:
return ''
df['Winner'] = df[['Vote A','Vote B']].apply(winner, axis=1)
哪个产量
Vote A Vote B Winner
1 4 B
3 2 A
1 5 B
