如何根據用戶輸入訪問資料幀行?假設我有一個包含世界各國的 csv 檔案。第一列是“大陸”,第二列是“國家”,第三列是“城市”。
如果我選擇某個國家,如何從“城市”中獲取所有值?
df = pd.read_csv("file.csv")
if continent == 'europe':
country = input("Choose a country:")
for index, row in enumerate(df.index):
city = df.values[index]
print(city)
使用此代碼,我只是列印出每個國家/地區的每個城市。如果我輸入德國,我只想得到德國城市。我在哪里放置“國家”變數在 for 回圈中?
uj5u.com熱心網友回復:
為此,您不需要 for 回圈。假設你的資料框列被稱為“大陸”、“國家”和“城市”,這樣的事情應該有效:
input_country= input("Choose a country:")
print (df.loc[df.country==input_country].city)
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/372793.html
下一篇:需要有關使用R清理資料的建議
