How to extract strings using Regex in pandas based on multiple if-conditions

Question

dframe = pd.DataFrame({'Num':['38-30','38-30','21-51','24-11','34-20'],
'Des':['Generates Vacuum','Pressure low','Ground Problem',
'Leak from Controller','Lock Unserviceable']})

In the above dataframe I want to extract specific strings from Des column based on the Num column. For instance, if Num is equal to 38-30 then extract Vacuum from Des and if it doesn't have Vacuum, then extract Pressure. There are multiple strings that I have to extract for each Num.

I am trying to use re.extract('generator|Pressure',re.I) but I don't know how to include the if statements as I mentioned above.

My output should look like this:

Yes. Is there a way that I can achieve that with Regex and If conditions? — Rakesh B
– Rakesh B, Commented May 27, 2020 at 21:44
@YOBEN's answer below involves regex and a vectorized IF statement — sammywemmy
– sammywemmy, Commented May 27, 2020 at 21:45

BENY · Accepted Answer · 2020-05-22 01:12:10Z

1

Use np.where with findall

np.where(df.Num.eq('38-30'),df.Des.str.findall('generator|vacuum',flags=re.IGNORECASE),'')

answered May 22, 2020 at 1:12

BENY

324k22 gold badges176 silver badges250 bronze badges

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

Collectives™ on Stack Overflow

How to extract strings using Regex in pandas based on multiple if-conditions

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