如何对含多级索引的Pandas DataFrame按一级索引归一化?
按多级索引一级分组的列归一化方案
直接用Pandas的groupby结合transform就能实现一行代码的简洁方案:
normalized_df = df.groupby(level=0).transform(lambda col: col / col.sum())
说明
groupby(level=0):指定按多级索引的第一级(即a、b分组)进行数据拆分。transform:对每个分组的每一列执行后续逻辑,并且自动保留原DataFrame的多级索引结构,无需手动调整结果格式。lambda col: col / col.sum():针对每个分组内的列,计算每个元素除以该列在当前分组的总和,完成归一化操作。
运行后得到的结果完全符合需求,所有值会以浮点数形式呈现(如1/5对应0.2,6/15对应0.4),同时保留原有的多级索引层级。
内容的提问来源于stack exchange,提问作者sds
相关产品推荐
相关产品推荐

