User input not working in pandas DataFrame

Question

here is the code!

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)]

in this code above the user will enter his choice of restaurant for instance user choose 'Bar', and then the code above will search the KEYWORD 'BAR' in the column name "collections" and return only those dataframe which consists of the keyword Bar in them. but this code is not working its returning me an empty dataframe. am using jupyter notebook

yibo · Accepted Answer · 2020-09-11 06:32:58Z

2

Problem

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

With quote, 'user' is a string in the above line. If nothing in Collections column contains "user" the result would be an empty DataFrame.

Solution

Remove the quote:

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

edited Sep 11, 2020 at 6:32

answered Sep 11, 2020 at 6:29

yibo

5274 silver badges14 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

bhansa · Accepted Answer · 2020-09-11 06:31:23Z

0

First thing, you have a typo in last line of your code, user is a variable.

Update the last line to:

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

Another approach:

dff[dff['Collections'] == user]

You might need to check for Case with this approach.

edited Sep 11, 2020 at 6:31

answered Sep 11, 2020 at 6:25

bhansa

7,5763 gold badges35 silver badges58 bronze badges

Collectives™ on Stack Overflow

User input not working in pandas DataFrame

2 Answers 2

Problem

Solution

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Problem

Solution

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related