如何根据用户输入访问 dataframe 行

Question

如何根据用户输入访问 dataframe 行？ 假设我有一个包含世界各国的 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 循环中？

Answer 1

为此，您不需要 for 循环。 假设您的 dataframe 列被称为“大陆”、“国家”和“城市”，这样的事情应该可以工作：

input_country= input("选择一个国家：") print (df.loc[df.country==input_country].city)