How do I filter for rows in a Pandas dataframe based on a column that has list as values?

Question

I have a Pandas dataframe where one of the columns stores lists of string literals. For example, my data looks like the following.

field1, field2
x1, ['tag1', 'tag2']
x2, ['tag1', 'tag3']

I want to get only the records in this dataframe whose field2 list contains tag1 or tag2. So, in the example above, only the record where field1=x1 should be returned.

How can I do this filtering on a Pandas dataframe?

Do you mean tag1 and tag2?

ALollz
– ALollz

2018-05-31 03:00:43 +00:00
Commented May 31, 2018 at 3:00 — ALollz
– ALollz, Commented May 31, 2018 at 3:00

harpan · Accepted Answer · 2018-05-31 02:58:35Z

1

You need to check for tag2 in the list of field2. Using .apply(), you can achieve this.

df[df.apply(lambda x: 'tag2' in x['field2'],axis=1)]

Output:

    field1  field2
0   x1      [tag1, tag2]

answered May 31, 2018 at 2:58

harpan

8,6412 gold badges22 silver badges40 bronze badges

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

Collectives™ on Stack Overflow

How do I filter for rows in a Pandas dataframe based on a column that has list as values?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related