如何利用另一DataFrame的信息为现有DataFrame创建新列
实现方案
你要的效果可以通过以下两种方法实现:
方法1:使用.map()实现(符合你的预期)
你可以先将df2转为Loc到City的映射字典,再对df1的Loc列做映射即可,代码如下:
# 构造映射字典:键为Loc值,值为对应城市名 loc_to_city = df2.set_index('Loc')['City'].to_dict() # 为df1新增City列 df1['City'] = df1['Loc'].map(loc_to_city)
该方案适合df2为维度表、Loc字段无重复的场景,执行效率高。
方法2:使用merge关联实现
你认为join不适用于该场景是误解,这种基于公共字段的左关联场景非常适合用merge实现,代码如下:
# 基于Loc字段做左关联,保留df1的所有行 df_result = df1.merge(df2, on='Loc', how='left')
如果后续需要从df2关联更多字段,该方案扩展性更强。
注意:如果df1中存在df2没有对应的Loc值,返回的City列会自动填充为NaN,两种方法均遵循该逻辑。
两种方法输出的结果都和你给出的期望输出完全一致。
内容的提问来源于stack exchange,提问作者reallycool123
相关产品推荐
相关产品推荐

