根據條件使用字典替換 python dataframe 中的列值

Question

我有以下 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

我有以下字典

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

我想使用 dict_1 替換 col2 中的值，但替換 col1 == 2 的行

所需的 output 如下

  col1     col2
0    4      USA
1    4  England
2    2    Tokyo

我嘗試了下面的方法，但它沒有做任何事情

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

Answer 1

切片時不要特別做inplace=True ：

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

Answer 2

另一種解決方案：

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

print(df)

印刷：

  col1     col2
0    4      USA
1    4  England
2    2    Tokyo