How to check if a series of strings is contained in a PANDAS DataFrame columns and assign that string as a new column in the row?

Question

I have a series of substrings that I want to check against a DataFrame column. For example:

SubStr = pd.series(['dog','cat','cow','fish'])

The DataFrame has a column called "String" where:

df['String'] = (['The dog went for a Walk.','The fish was in the lake.','The dog was barking'])

I want to add a column to the DataFrame that contains the SubStr found in 'String' for that row or just NaN if none of them are found. In my example the new column should contain:

df['StrLookUp'] = ['dog','fish','dog']

In my search research I was able to find examples of where elements of a series are searched for any items in a list, but none of them returned the specific element that was found.

A.Kot · Accepted Answer · 2017-10-24 18:41:47Z

2

Use regex:

import re

pattern= '|'.join(['dog','cat','cow','fish'])

df['StrLookUp'] = [re.findall(pattern, i) for i in df['String']]

answered Oct 24, 2017 at 18:41

A.Kot

7,9732 gold badges24 silver badges24 bronze badges

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

Collectives™ on Stack Overflow

How to check if a series of strings is contained in a PANDAS DataFrame columns and assign that string as a new column in the row?

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