Filter using keyword Pandas Dataframe Python

Question

I have data stored in the data frame. Now I want to filter out the data within specific columns using Keywords. For Example: let my dataframe be:

    customer    product         state
0   shyam       pen drive       delhi
1   suresh      hard disk       punjab
2   shanu       USB pen drive   haryana
3   rohan       pen drive 16gb  uttarpradesh

Now I want to filter this dataframe in column 'product' with values containing 'pen' only. so the result should look like:

    customer    product         state
0   shyam       pen drive       delhi
2   shanu       USB pen drive   haryana
3   rohan       pen drive 16gb  uttarpradesh

Every suggestion would be appreciated.

score 2 · Accepted Answer · answered Oct 08 '20 at 15:58

2

Use this:

df[df['product'].str.contains('pen')]

Output:

  customer         product         state
0    shyam       pen drive         delhi
2    shanu   USB pen drive       haryana
3    rohan  pen drive 16gb  uttarpradesh

.str, string accessor, can be slow, use list comprehension for better performance:

df[['pen' in i for i in df['product']]]

answered Oct 08 '20 at 15:58

Scott Boston

114,762
11
99
130

1

@ShubhamSharma Thanks! – Scott Boston Oct 08 '20 at 16:15
1

Thanks! Also is there any way to find using multiple keywords, like: `df[df['product'].str.contains('pen','hard','something_else')]` – Sunny Singh Oct 08 '20 at 17:51
1

Yes. use regex `.str.contains('pen|hard|something_else')`. – Scott Boston Oct 08 '20 at 18:45

score 0 · Answer 2 · answered Oct 08 '20 at 16:13

0

df2 = df.loc[df.product.str.contains('pen')]

answered Oct 08 '20 at 16:13

lager

49
6

Filter using keyword Pandas Dataframe Python

2 Answers2