如何强制pandas MultiIndex在索引值一致时按共有层级广播运算

问题原因 #

pandas 对 MultiIndex 执行二元运算时，优先按索引元组的值顺序和内容匹配：如果两个 Series 的 MultiIndex 长度相同、对应位置的元组完全一致，无论层级名是否不同，都会直接按位置对齐运算，不会触发跨层级广播。只有当索引元组无法完全匹配时，才会自动识别共有层级执行广播。

解决方案 #

方法1：关联法（最直观，适合复杂业务逻辑） #

通过merge将两个 Series 的索引展开为字段，按共有层级关联生成笛卡尔积后再运算：

import pandas as pd

# 转换为带索引字段的DataFrame
s_df = s.reset_index()
u_df = u.reset_index()

# 按共有层级Common关联，自动生成笛卡尔积
merged = pd.merge(s_df, u_df, on="Common")

# 重新设置三层索引后执行按位与运算
result = merged.set_index(["Common", "Level s", "Level u"]).eval("s & u")

方法2：手动构造索引法（性能更优，适合大数据量场景） #

主动构造两个索引的笛卡尔积作为最终索引，将两个Series重排到新索引上再运算：

# 构造三层笛卡尔积索引
new_index = pd.MultiIndex.from_product(
    [
        s.index.get_level_values("Common").unique(),
        s.index.get_level_values("Level s").unique(),
        u.index.get_level_values("Level u").unique()
    ],
    names=["Common", "Level s", "Level u"]
)

# 按共有层级重排广播
s_align = s.reindex(new_index, level=["Common", "Level s"])
u_align = u.reindex(new_index, level=["Common", "Level u"])

# 执行按位与运算
result = s_align & u_align

结果验证 #

两种方法得到的result结构如下，和预期广播效果一致：

Common  Level s  Level u
Foo     a        a           0
                 b           0
        b        a           0
                 b           0
dtype: int64

（注：按位与结果为0是因为示例值1&100=0、2&200=0，符合位运算规则）

内容的提问来源于stack exchange，提问作者Olof Ahlén