Получение максимального значения DataFrame

Question

У меня есть датафрейм с индексами названий стран и столбцами медалей. Я хочу получить название страны с наибольшим количеством золотых медалей. Я пробовал это:

def answer_one():
    x= df[df['Gold.2']==df['Gold.2'].max()]
    return x.index
answer_one()

Я хочу получить только строку, которая является названием страны, но вместо этого я продолжаю получать это

Index(['United States'], dtype='object')

Answer 1

Если вам нужно конкретное значение, я бы использовал следующий код:

def answer_one():
    x= df[df['Gold.2']==df['Gold.2'].max()]
    return x.index.values[0]
answer_one()

Это вернет первую максимальную страну. Если вы хотите получить массив всех максимальных стран:

def answer_one():
    x= df[df['Gold.2']==df['Gold.2'].max()]
    return x.index.values
answer_one()

Answer 2

def answer_one():
    x= df[df['Gold.2']==df['Gold.2'].max()]
    return x.index.values[0]
answer_one()