熊猫“<class‘class’>类型”

Question

我正在用奥运会的数据编写一个程序。

所提出的问题是：

哪个国家在夏季奥运会上获得的金牌最多？

答案是：

美国

以下是一些可供参考的变量和数据点：

# Summer  Gold  Silver  Bronze  Total medals
United Arab Emirates 8  1   0   0   1
United States   26  976 757 666 2399
Uruguay 20  2   2   6   10

下面的函数应该返回一个字符串值，但它返回的是一个熊猫数据帧。

def answer_one():
    return df.loc[df['Gold'] == df['Gold'].max()]

我如何纠正这个函数？

Answer 1

取而代之的是这样做：

def answer_one():
    return list(df.loc[df['Gold'] == df['Gold'].max()].index)

如果有联系的话，这应该会给你一个拥有最多金牌的国家名单。如果没有联系，它只会给出一个包含一个条目的列表:在您的例子中，结果将是['US']。

Answer 2

如果您的索引列是国家名称，那么这将起到以下作用：

df.loc[df['Gold'].idxmax()]

Answer 3

你可以试试这个。对我来说很管用

df.Gold.idxmax()

干杯