根据其他三列的多数值设置pandas dataframe winner列值

Question

我这有熊猫df

id  Vote1     Vote2      Vote3 
123 Positive  Negative   Positive
223 Positive  Negative   Neutral 
323 Positive  Negative   Negative  
423 Positive  Positive             

我想添加另一个名为winner的列 这将被设置为任何多数票，如果票数相等，则将设置第一票，如id=223所示

所以结果df应该是

^{pr2}$

这可能与 Update Pandas Cells based on Column Values and Other Columns

Answer 1

你可以这样排：

import pandas as pd
import numpy as np

# Create the dataframe
df = pd.DataFrame()
df['id']=[123,223,323,423]
df['Vote1']=['Positive']*4
df['Vote2']=['Negative']*3+['Positive']
df['Vote3']=['Positive','Neutral','Negative','']

mostCommonVote=[]
for row in df[['Vote1','Vote2','Vote3']].values:
    votes, values = np.unique(row, return_counts=True)
    if np.all(values<=1):
            mostCommonVote.append( row[0] )
    else:
        mostCommonVote.append( votes[np.argmax(values)] )

df['Winner'] = mostCommonVote

结果：

^{pr2}$

它可能不是最优雅的解决方案，但它相当简单。它使用numpy函数unique，该函数可以返回行中每个唯一字符串的计数。在