python - Pandas :用条件替换列中的值

Question

我有数据框

city_reg     city_live   reg_region    live_region 
 Moscow         Tver        77            69
 Tambov         Tumen'      86            86

如果 reg_region == live_region，我需要将 city_reg 中的值替换为 city_live 中的值

我尝试使用

df.loc[df.reg_region == df.live_region, 'city_reg'] = df['city_live']

但它返回

ValueError: cannot reindex from a duplicate axis

我该如何解决？

Answer 1

使用mask或 numpy.where与重复索引一起工作非常好:

#create duplicated indices for test
df.index = [0,0]
print (df)
  city_reg city_live  reg_region  live_region
0   Moscow      Tver          77           69
0   Tambov    Tumen'          86           86

df['city_reg'] = df['city_reg'].mask(df.reg_region == df.live_region,  df['city_live'])

或者:

df['city_reg'] = np.where(df.reg_region == df.live_region,  df['city_reg'], df['city_live'])

---

print (df)
  city_reg city_live  reg_region  live_region
0   Moscow      Tver          77           69
0   Tumen'    Tumen'          86           86