Pandas - Extract value from column in data frame

Question

I have a pandas data frame with really long text in a column. I wanted to select all columns that contain ABC. I was able to do this using the following

 df[df['Column'].str.contains('ABC', na=False)]

What I want to do after that is extract all values from this field that contain the prefix and the next 5 letters. S.So after finding a column, I would want to get ABC1234 or ABC7899.

I hope this makes sense.

Chris · Accepted Answer · 2022-09-26 17:01:11Z

2

You can use str.extract with a regular expression that says to capture any time it sees ABC with 5 following digits

df = pd.DataFrame({'Column':['ABC12345 is in this column', 'Not in this one CCD11111','Also in this one ABC99882']})
df['capture'] = df.Column.str.extract('(ABC\d{5})')
df.dropna(inplace=True)
print(df)

Output

                      Column   capture
0  ABC12345 is in this column  ABC12345
2   Also in this one ABC99882  ABC99882

answered Sep 26, 2022 at 17:01

Chris

16.3k3 gold badges26 silver badges41 bronze badges

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

1 Comment

Kulwant Over a year ago

Thanks so much. I was really stuck on this.

Collectives™ on Stack Overflow

Pandas - Extract value from column in data frame

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related