Extract string from a dataframe comparing to a list
You can use extract
with join
all values in List
by |
what means or
in regex
:
List1 = ['Three has' , 'Mail' , 'Done' , 'Game' , 'Time has come']
df['Col 3'] = df['Col 2'].str.extract("(" + "|".join(List1) +")", expand=False)
print (df)
Col 1 Col 2 Col 3
0 1 The date NaN
1 2 Three has come Three has
2 3 Mail Sent Mail
3 4 Done Deal Done
Another solution:
List1 = ['Three has' , 'Mail' , 'Done' , 'Game' , 'Time has come']
df['Col 3'] = df['Col 2'].apply(lambda x: ''.join([L for L in List1 if L in x]))
df['Col 3'] = df['Col 3'].mask(df['Col 3'] == '')
print (df)
Col 1 Col 2 Col 3
0 1 The date NaN
1 2 Three has come Three has
2 3 Mail Sent Mail
3 4 Done Deal Done