CODEWITHSUNDEEP
pythonpython-2.7pandas
rurp
rurp

Reputation: 1446

How to filter a DataFrame column of lists for those that contain a certain item

If I want to filter a column of strings for those that contain a certain term I can do so like this:

df = pd.DataFrame({'col':['ab','ac','abc']})
df[df['col'].str.contains('b')]

returns:

   col
0   ab
2  abc

How can I filter a column of lists for those that contain a certain item? For example, from

df = pd.DataFrame({'col':[['a','b'],['a','c'],['a','b','c']]})

how can I get all lists containing 'b'?

         col
0     [a, b]
2  [a, b, c]

Upvotes: 20

Views: 21100

Answers (1)

chrisb
chrisb

Reputation: 52246

You can use apply, like this.

In [13]: df[df['col'].apply(lambda x: 'b' in x)]
Out[13]: 
         col
0     [a, b]
2  [a, b, c]

Although generally, storing lists in a DataFrame is a bit awkward - you might find some different representation (columns for each element in the list, MultiIndex, etc) that is easier to work with.

Upvotes: 35

Related Questions