熊貓“類型<class 'str'> ”

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

如果 Country name 是您的索引列，這應該可以解決問題：

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

Answer 3

你可以試試這個。 它對我有用

df.Gold.idxmax()

干杯

Answer 4

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