Pandas多级索引Series分组均值计算后无法对齐相减如何解决？

运算报错的核心原因：b是仅包含level=0级的单级索引Series，和两级MultiIndex的a做算术运算时无法自动对齐匹配，导致广播失败。

你可以通过以下两种方法实现需求： #

方案1：使用transform直接保留原索引结构（推荐） #

groupby.transform方法计算分组聚合结果后，会自动将结果广播匹配原数据的索引结构，无需手动处理对齐：

import pandas as pd

a = pd.Series({(2,1): 3., (1,2):4.,(2,3):4.})
r = a - a.groupby(level=0).transform('mean')
print(r)

输出结果完全符合预期：

1  2    0.0
2  1   -0.5
   3    0.5
dtype: float64

方案2：手动对齐已计算的分组均值 #

如果你已经提前计算好了单级索引的分组均值b，可以通过reindex方法指定level参数完成索引对齐后再做运算：

import pandas as pd

a = pd.Series({(2,1): 3., (1,2):4.,(2,3):4.})
b = a.groupby(level=0).mean()
r = a - b.reindex(a.index, level=0)
print(r)

输出结果和方案1完全一致。

内容的提问来源于stack exchange，提问作者silgon