python中idx函数_python中idxmax_pandas自定义idxmax（）函数，具有特殊的相等处理

我正在处理一个csv文件，里面装满了选举数据。我的原始样本可以表示为：city party1 party2 party3

0 city1 50 107 114

1 city2 181 323 326

2 city3 26 28 75

3 city4 32 47 59

4 ciy5 8 21 21

我使用pandas的idxmax()函数创建了一个名为“winner”的新列，如下所示：

^{pr2}$

我的目标是确定哪个政党在每个城市中处于第一位。结果如下：city party1 party2 party3 winner

0 city1 50 107 114 party3

1 city2 181 323 326 party3

2 city3 26 28 75 party3

3 city4 32 47 59 party3

4 ciy5 8 21 21 party2

胜利者的最后一个raw的值是错误的，因为party2和party3的分数相同。在

是否可以在函数idxmax考虑两个值相等并给出“equality”时包含一个异常？在