filter rows based on multiple strings with Pandas

Asked Jun 03 '18 at 08:14

Active Jun 03 '18 at 08:14

Viewed 31 times

If I have a Pandas DataFrame like this :

id     fruits
01     Apple, Apricot
02     Apple, Banana, Clementine, Pear
03     Orange, Pineapple, Pear
04

How can i filter rows which contains Apricot, Orange and Pineapple and get following dataframe? Thanks.

id     fruits
01     Apple, Apricot
03     Orange, Pineapple, Pear

asked Jun 03 '18 at 08:14

ah bon

Those are strings? – cs95 Jun 03 '18 at 08:17
Have your fruits in a list of strings and then use the solution above – cs95 Jun 03 '18 at 08:18
Yes, i think so. – ah bon Jun 03 '18 at 08:18
Thanks. Do you know deal with this error? ValueError: cannot index with vector containing NA / NaN values. I try dropna, it doesn't work out. – ah bon Jun 03 '18 at 08:30
@ahbon add parameter `na=False` to `str.contains` – jezrael Jun 03 '18 at 08:41
Thanks guys. This one seems worked perfectly. df[(df['fruits'].str.contains('Apricot|Orange|Pineapple', na=False))] – ah bon Jun 03 '18 at 08:52

0 Answers0