Can I make a Python if condition using Regex on Pandas column to see if it contains something and then create a new column to hold it

Question

Say I have data like this

	Account
1	Kevin (1234567)
2	Buzz (7896345)
3	Snakes (5438761)
4	Marv
5	Harry (9083213)

I want to use an if condition to search to see if the account number exists at the end of the name in the account column, if it does split the account number off and put it in a new column, if not pass and go on to the next Account.

Something like this although it does not work

dataset.loc[dataset.Account.str.contains(r'\(d+')], 'new'=dataset.Account.str.split('',n=1, expand=True)
dataset["Account_Number"] = new[1]

@lucas_7_94 yes I was thinking just leaving it Na

Tanner Ormanoski
– Tanner Ormanoski

2022-10-19 15:48:19 +00:00
Commented Oct 19, 2022 at 15:48 — Tanner Ormanoski
– Tanner Ormanoski, Commented Oct 19, 2022 at 15:48
To match digits, you need \d+, not d+.

Wiktor Stribiżew
– Wiktor Stribiżew

2022-10-19 15:50:38 +00:00
Commented Oct 19, 2022 at 15:50 — Wiktor Stribiżew
– Wiktor Stribiżew, Commented Oct 19, 2022 at 15:50

Andrej Kesely · Accepted Answer · 2022-10-19 15:47:41Z

1

Try:

df["Account number"] = df["Account"].str.extract(r"\((\d+)\)$")
df["Account"] = df["Account"].str.replace(r"\s*\(\d+\)$", "", regex=True)
print(df)

Prints:

  Account Account number
1   Kevin        1234567
2    Buzz        7896345
3  Snakes        5438761
4    Marv            NaN
5   Harry        9083213

answered Oct 19, 2022 at 15:47

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

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

Comments

Naveed · Accepted Answer · 2022-10-19 15:47:08Z

1

here is one way to do it

# split the account on ( and create two columns
df[['Account','Account Number']]= df['Account'].str.split('(', expand=True)

#replace the trailing ) with empty string
df['Account Number']=df['Account Number'].str.replace(r'\)','', regex=True ).str.strip()
df
dfdf

        Account     Account Number
0   1   Kevin              1234567
1   2   Buzz               7896345
2   3   Snakes             5438761
3   4   Marv                  None
4   5   Harry              9083213

answered Oct 19, 2022 at 15:47

Naveed

11.7k2 gold badges16 silver badges21 bronze badges

Collectives™ on Stack Overflow

Can I make a Python if condition using Regex on Pandas column to see if it contains something and then create a new column to hold it

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related