python - 用户输入在 pandas DataFrame 中不起作用

Question

这是代码！

dff=pd.read_csv('file.csv')

#finding those restaurant which have bar's at night
user=input('enter here')
dff[dff['Collections'].str.contains('user',na=False)]

在上面的代码中，用户将输入他选择的餐厅，例如用户选择'Bar'，然后上面的代码将在列名“collections”中搜索关键字“BAR”，并仅返回包含关键字 Bar 的那些数据框在他们中。但是这段代码不起作用，它返回给我一个空的数据框。我正在使用 jupyter 笔记本

标签： pythonpandasdataframe

Answer 1

问题

dff[dff['Collections'].str.contains('user', na=False)]

加上引号，'user'是上面一行中的一个字符串。如果 Collections 列中没有任何内容包含“用户”，则结果将是一个空的 DataFrame。

解决方案

删除引号：

dff[dff['Collections'].str.contains(user, na=False)]