如何合并带MultiIndex的Pandas DataFrame,实现低维数据自动重复匹配
解决方法
你需要的是让低维DataFrame的条目自动广播匹配到高维DataFrame的每个frame层级,Pandas本身就支持这种操作,不需要手动复制行,下面是两种简单的实现方式:
方法一:直接用join(推荐)
因为两个DataFrame的索引有共同的层级(individuals, bodyparts, coords),join会自动将低维数据广播到高维的每个frame条目:
- 先给r3p的列重命名,避免和d2nlt的
value列冲突:
r3p = r3p.rename(columns={'value': 'r3p_value'})
- 直接执行join操作:
merged_df = d2nlt.join(r3p)
这样得到的merged_df里,每个individuals-bodyparts-coords对应的所有frame行,都会自动填充对应的r3p_value值。
方法二:用merge配合索引重置
如果习惯用merge,可以先把MultiIndex转为普通列,匹配后再恢复索引:
- 重置两个DataFrame的索引为列:
d2nlt_flat = d2nlt.reset_index() r3p_flat = r3p.reset_index()
- 只按共同的列(
individuals, bodyparts, coords)做左连接:
merged_flat = d2nlt_flat.merge(r3p_flat, on=['individuals', 'bodyparts', 'coords'], how='left')
- 恢复原来的MultiIndex:
merged_df = merged_flat.set_index(['individuals', 'bodyparts', 'coords', 'frame'])
为什么你之前的操作无效?
你之前的merge把frame也加入了on参数,但r3p里没有frame列,导致无法匹配。正确的做法是只基于两者共有的层级/列做匹配,Pandas会自动将低维数据重复填充到高维的每个frame条目上。
内容的提问来源于stack exchange,提问作者Andrew Spence
相关产品推荐
相关产品推荐

