python - 根据 Pandas 中另一列的值添加新列(python)

Question

给定一个看起来像这样的数据框

Vote A Vote B
   1     4      
   3     2      
   1     5      

我想添加一个名为 Winner 的新列，用于比较 Vote A 和 Vote B 两列之间的值并指定获胜者。

Winner
B
A
B

我如何在 Pandas 中执行此操作？

Answer 1

你可以使用

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