[Solved] Compare strings of a column in a dataframe with a set of words in a list

by


Ok, let’s assume we have a dataframe data and list negative_words like this:

data = pd.DataFrame({
    'Tweets' : ['This is bad', 'This is terrible', 'This is good', 'This is great'],
})

negative_words = ['bad', 'terrible']

We can then do something like:

1) We can use a lambda function with any:

# create lambda with any:
data['Negative'] = data.apply(lambda x: True if any(word in x.Tweets for word in negative_words) else False, axis=1)

And will get:

             Tweets  Negative
0       This is bad      True
1  This is terrible      True
2      This is good     False
3     This is great     False

7

solved Compare strings of a column in a dataframe with a set of words in a list