Create new column in pandas if list contain any string of another sublist

Question

Let's say I have this df:

id words
1  ['apple', 'banana', 'grape']
2  ['apple', 'banana', 'chocolate']
3  ['grape', 'popcorn', 'chocolate']
4  ['grape', 'apple', 'popcorn']

How can I create a new column checking if 'chocolate' or 'popcorn' is inside each row-list. I've tried something like this but it didn't work:

sublist = ['chocolate', 'popcorn']

df['boolean_check'] = df['words'].isin(sublist).any()

also tried:

 df['boolean_check'] = np.where(df['words'].isin(sublist).any(), True, False)

expected result:

id words                             boolean_check
1  ['apple', 'banana', 'grape']      False
2  ['apple', 'banana', 'chocolate']  True
3  ['grape', 'popcorn', 'chocolate'] True
4  ['grape', 'apple', 'popcorn']     True

BENY · Accepted Answer · 2021-05-20 20:33:12Z

1

Try with explode

np.where(df['words'].explode().isin(sublist).any(level=0), True, False)

answered May 20, 2021 at 20:33

BENY

324k22 gold badges176 silver badges250 bronze badges

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

1 Comment

SeaBean Over a year ago

No need for np.where() ?

Collectives™ on Stack Overflow

Create new column in pandas if list contain any string of another sublist

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related