select a partial string from a pandas DataFrame save in a column

Question

I need to extract the values from a column in pandas df and save those values in a separate column. I need to grab the strings that follow a specific pattern. My data looks like this:

name        
ADV | FashionWeek.ab
ADV | FashionWeek
ADV | ESPN.arb
ADV | ESPN.ob
ADV | ESPN

The desired output is:

name                        clean_name  
ADV | FashionWeek.ab       FashionWeek
ADV | FashionWeek          FashionWeek
ADV | ESPN.arb             ESPN
ADV | ESPN.ob              ESPN
ADV | ESPN                 ESPN

So I have to grab everything that comes after ADV | and before . I tried:

new_df["clean_name"] = df["name"].split('|')[1].lstrip().split('.')[0]

error: AttributeError: 'DataFrame' object has no attribute 'split'

Ani · Accepted Answer · 2021-01-15 17:51:44Z

2

You have the string logic right, you were just applying the split function to the dataframe object and not the items inside the column. To do this, you can use the .apply function:

df["clean_name"] = df["name"].apply(lambda x: x.split('|')[1].lstrip().split('.')[0])

answered Jan 15, 2021 at 17:51

Ani

5863 silver badges4 bronze badges

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

Collectives™ on Stack Overflow

select a partial string from a pandas DataFrame save in a column

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