如何重新索引Pandas数据帧的列多索引？

Question

我有一个Pandas DataFrame，列上有MultiIndex（假设有3个级别）：

MultiIndex(levels=[['BA-10.0', 'BA-2.5', ..., 'p'], ['41B004', '41B005', ..., 'T1M003', 'T1M011'], [25, 26, ..., 276, 277]],
           labels=[[0, 0, 0, ..., 18, 19, 19], [4, 5, 6,..., 14, 12, 13], [24, 33, 47, ..., 114, 107, 113]],
           names=['measurandkey', 'sitekey', 'channelid'])

当我通过第一级并生成DataFrame的子集时：

^{pr2}$

Columns index确实丢失了它的第一级，但是MultiIndex仍然包含对子级中所有其他键的引用，即使它们在子集中丢失了。在

MultiIndex(levels=[['41B004', '41B005', '41B006', '41B008', '41B011', '41MEU1', '41N043', '41R001', '41R002', '41R012', '41WOL1', '41WOL2', 'T1M001', 'T1M003', 'T1M011'], [25, 26, 27, 28, 30, 31, 32, 3, ....
           labels=[[4, 5, 6, 7, 9, 10], [24, 33, 47, 61, 83, 98]],
           names=['sitekey', 'channelid'])

如何强制subdf只使用存在的键更新其列MultiIndex？在

Answer 1

def cluster(df):
    for key in df.columns.levels[0]:
        d = df[key]
        d.columns = pd.MultiIndex.from_tuples(d.columns.to_series())
        yield d