Extracting Specific Text From column in dataframe in pandas

Question

I have a pandas dataframe with a column, which I need to extract the word with [ft,mi,FT,MI] of the state column using regular expression and stored in other column.

 df1 = {
    'State':['Arizona 4.47ft','Georgia 1023mi','Newyork 2022 NY 74.6 FT','Indiana 747MI(In)','Florida 453mi FL']}

Expected output

               State  Distance
0     Arizona 4.47ft  4.47ft
1     Georgia 1023mi  1023mi
2  Newyork NY 74.6ft  74.6ft
3  Indiana 747MI(In)   747MI
4   Florida 453mi FL   453mi

Would anyone please help?

Shubham Sharma · Accepted Answer · 2022-02-17 15:57:40Z

1

Build a regex pattern with the help of list l then use str.extract to extract the occurrence of this pattern from the State column

l = ['ft','mi','FT','MI']
df1['Distance'] = df1['State'].str.extract(r'(\S+(?:%s))\b' % '|'.join(l))

                    State Distance
0          Arizona 4.47ft   4.47ft
1          Georgia 1023mi   1023mi
2  Newyork 2022 NY 74.6FT   74.6FT
3       Indiana 747MI(In)    747MI
4        Florida 453mi FL    453mi

answered Feb 17, 2022 at 15:57

Shubham Sharma

71.8k6 gold badges26 silver badges58 bronze badges

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

2 Comments

Learner Over a year ago

I forgot to mention that i have some spaces after 74.6 FT

Shubham Sharma Over a year ago

Check with: df1['State'].str.extract(r'(\S+\s?(?:%s))\b' % '|'.join(l))

Collectives™ on Stack Overflow

Extracting Specific Text From column in dataframe in pandas

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related