如何根据用户输入访问数据框行答案

【问题标题】：How to access dataframe rows based on user input如何根据用户输入访问数据框行
【发布时间】：2022-01-09 12:04:04
【问题描述】：

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

【问题讨论】：

你能展示你的数据框样本吗？你有行的多索引吗？或者，你有默认的范围索引和三列吗？
我目前正在使用手机，所以我现在无法显示示例，但我有默认范围索引和三列。
df.query('country == @country')['city'].to_list() 不需要循环。在输入行之后试试这个。

标签： python pandas csv file user-input

【解决方案1】：

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

input_country= input("Choose a country:")
print (df.loc[df.country==input_country].city)

【讨论】：

非常感谢！帮了我很多:)