Pandas replace substring from dict of values

Question

Below is sample Dataframe

data = {'A': ['hi UK','hi IN','hi US']}

df = pd.DataFrame(data)

I wanted to update UK, IN values for A column from matching dict below

abs = {'U': 'UK -- extra', 'UK': 'test Kingdom', 'IN':'India'}

then I used replace function (pandas.DataFrame.replace)

df['A'] = df['A'].replace(to_replace = abs, regex=True)
print(df)

                           A
0  hi test Kingdom -- extraK
1                   hi India
2  hi test Kingdom -- extraS

Its replacing U first by UK -- extra and again UK with test kingdom so final result is hi test Kingdom -- extraK idially it should give test kingdom

The expected output is

                 A
0  hi test Kingdom
1         hi India
2            hi US

Am i missing any thing or is there anyway to achieve above result.

Thanks in advance.

jezrael · Accepted Answer · 2018-07-26 11:28:01Z

2

I think word boundary should help here for match UK, US and no U:

data = {'A': ['hi UK','hi IN','hi US']}

d = {'U': 'UK -- extra', 'UK': 'test Kingdom', 'IN':'India'}

d = {r'\b{}\b'.format(k):v for k, v in d.items()}
df = pd.DataFrame(data)

df['A'] = df['A'].replace(to_replace = d, regex=True)
print(df)
                 A
0  hi test Kingdom
1         hi India
2            hi US

answered Jul 26, 2018 at 11:28

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

Pandas replace substring from dict of values

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