Isolate a word from a long string

[nicocorico]

Hello everyone,

I am trying to isolate specific words out of a string if present but until now with no success.
If in the string I have 'irg', 'swopt', 'fvol' or 'swol' then I want to write the word found in a specific column. The string I look at have specific caracters.
Example: http:\dsddf;dfdfdf-dfdfd.irg/fdfdfd

I have tried the following code but with no success as it could found none of the word listed above and therefore as a result I have 'Category' column empty.

category = r"irg|swopt|fvol|swvol"
df_susanoo.insert(22, 'Category', df_susanoo['delivery_detail'].str.findall(category).apply(lambda x: x if len(x) >1 else ['not found']).str[-1])

['delivery_detail'] column contains the string to look at.

Would you see what might be wrong?

Thank you in advance for your help ,
Nicolas

[ibreeden]

Well that is an impressive line of code you are showing us. But what does it mean? We could not tell as we don't know how df_susanoo is defined and what the contents are. And does the program give you an error message? I bet it does. Show us the complete error message.

You should build your line of code step by step. Start with:

import re
example_data = "http:\dsddf;dfdfdf-dfdfd.irg/fdfdfd"
category = r"irg|swopt|fvol|swvol"
re.findall(category, example_data)

Output:
['irg']

Is this what you wanted? Then continue to add statements until you get what you want.
Let us know the result of your investigation.

If you don't succeed then make a small sample of your program that we can run and shows exactly what goes wrong. Don't forget to include the full error message if you get one.

[nicocorico]

Thank you for your reply! I tried and it works perfectly!

category = r"irg|swopt|fvol|swvol"

for i in tqdm(range(0,len(df_susanoo))):
    if len(re.findall(category, df_susanoo.iloc[i,1])) != 0:
        df_susanoo.iloc[i,22] = re.findall(category, df_susanoo.iloc[i,1])[0]
    else: df_susanoo.iloc[i,22] = ""

df_susanoo.iloc[i,1] is corresponding to the column where there is the string to search on.