以字符串形式返回索引值

Question

我試圖以字符串形式返回值的索引。我在這裏看到的其他問題已經將索引作爲列表返回。

時引發的錯誤是： 你返回類型的變量，我們預計一類

我的代碼：

String_to_be_returned=(df['Column'].index[df['Column']==2]) 

例子：

當我打印String_to_be_returned，我得到這個：

Index（['United States']，dtype ='object'，name ='Country Name'）

Answer 1

我認爲你需要添加[0]爲index選擇第一個值是array：

String_to_be_returned= df[df['Column']==2].index[0] 

樣品：

df = pd.DataFrame({'Column':[1,2,3], 
        'Column1':[4,5,6] 
        }, index=['Slovakia','United States','Mexico']) 

print (df) 
       Column Column1 
Slovakia   1  4 
United States  2  5 
Mexico    3  6 

String_to_be_returned= df[df['Column']==2].index[0] 
print (String_to_be_returned) 
United States 

---

String_to_be_returned= df.index[df['Column']==2][0] 
print (String_to_be_returned) 
United States