Replace column values in python dataframe using dictionary based on condition

Question

I have below dataframe

df = pd.DataFrame({'col1': {0: '4', 1: '4', 2: '2'},'col2': {0: 'USA', 1: 'England', 2: 'Japan'}})

>>> df
  col1     col2
0    4      USA
1    4  England
2    2    Japan

and I have below dictionary

dict_1 = {"USA" : 'Washington',"Japan" : 'Tokyo',"England" : 'London'}

I want to replace values in col2 using dict_1 but replace in rows where col1 == 2

Desired output is as below

  col1     col2
0    4      USA
1    4  England
2    2    Tokyo

I tried below method but it doesnt do anything

df.loc[df['col1'] == '2', 'col2'].replace(dict_1,inplace=True)

Quang Hoang · Accepted Answer · 2022-09-22 20:07:58Z

3

Don't do inplace=True specially when you slice:

df.loc[df['col1']=='2', 'col2'] = df.loc[df['col1'] == '2', 'col2'].replace(dict_1)

Output:

  col1     col2
0    4      USA
1    4  England
2    2    Tokyo

answered Sep 22, 2022 at 20:07

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Comments

Andrej Kesely · Accepted Answer · 2022-09-22 20:09:58Z

2

Another solution:

m = df.col1.eq("2")
df.loc[m, "col2"] = df.loc[m, "col2"].map(dict_1)

print(df)

Prints:

  col1     col2
0    4      USA
1    4  England
2    2    Tokyo

answered Sep 22, 2022 at 20:09

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

1 Comment

Quang Hoang Over a year ago

Just a quick note that map and replace behave differently on not-found items.

Collectives™ on Stack Overflow

Replace column values in python dataframe using dictionary based on condition

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related