python - 使用 idxmax 求最大值

Question

我正在尝试计算夏季金牌数和冬季金牌数相对于金牌总数的最大差异。问题是我只需要考虑在夏季和冬季都至少赢得 1 枚金牌的国家。

Gold: Count of summer gold medals

Gold.1: Count of winter gold medals

Gold.2: Total Gold

这是我的数据示例:

            Gold    Gold.1  Gold.2  ID  diff gold %
Afghanistan 0       0       0       AFG NaN
Algeria     5       0       5       ALG 1.000000
Argentina   18      0       18      ARG 1.000000
Armenia     1       0       1       ARM 1.000000
Australasia 3       0       3       ANZ 1.000000
Australia   139     5       144     AUS 0.930556
Austria     18      59      77      AUT 0.532468
Azerbaijan  6       0       6       AZE 1.000000
Bahamas     5       0       5       BAH 1.000000
Bahrain     0       0       0       BRN NaN
Barbados    0       0       0       BAR NaN
Belarus     12      6       18      BLR 0.333333

这是我的代码，但它给出了错误的答案:

def answer():
    Gold_Y = df2[(df2['Gold'] > 1) | (df2['Gold.1'] > 1)]
    df2['difference'] = (df2['Gold']-df2['Gold.1']).abs()/df2['Gold.2']
    return df2['diff gold %'].idxmax()
   
answer()  

Answer 1

添加正确的(您的)函数和变量名称后尝试此代码。我是 Python 新手，但我认为问题是您必须在第 4 行 (df1['difference']) 中使用相同的变量，然后将方法 (.idxmax()) 添加到末尾。我认为您也不需要该函数的第一行代码，因为您不使用局部变量(Gold_Y)。仅供引用 - 我不认为我们正在使用相同的数据集。

def answer_three():
    df1['difference'] = (df1['Gold']-df1['Gold.1']).abs()/df1['Gold.2']
    return df1['difference'].idxmax()

answer_three()