Pandas: How to filter column list by value?

Question

I have a Pandas DataFrame that stores lists in one of its columns:

>>> import pandas as pd
>>> d = [{'name': 'john', 'properties': ['a','b']},
...      {'name': 'mary', 'properties': ['a','c']}]
>>> df = pd.DataFrame(d)
>>> df
   name properties
0  john     [a, b]
1  mary     [a, c]

How can I filter by list membership? For example, list all rows that have a 'c' in their properties column.

I know I could explode the properties column:

df.explode('properties')

but I'd prefer to keep it as a list.

Michael Gardner · Accepted Answer · 2019-10-11 22:20:19Z

1

You can use map.

df[df.properties.map(lambda x: 'c' in x)]

answered Oct 11, 2019 at 22:20

Michael Gardner

1,8231 gold badge14 silver badges14 bronze badges

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

Collectives™ on Stack Overflow

Pandas: How to filter column list by value?

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