将pandas multiindex折叠为单个索引

Question

我有一个多索引的Pandas数据框,如下所示:

如何将三层索引合并为一个索引？也就是说,我想把(1987,1,2)变成pd.datetime(1987,1,2).我更喜欢使用df.index.map的矢量化方法.这里的代码可以创建数据框的顶部:

df = pd.DataFrame(
    {'3 months': [1, 2, 3, 4, 5]},
    index=[
        [1987, 1987, 1987, 1987,1987],
        [1,1,1,1,1],
        [2,5,6,7,8]
    ]
)

Answer 1

最简单的解决方案 pd.Index.map

df.set_index(df.index.map(lambda t: pd.datetime(*t)))

            3 months
1987-01-02         1
1987-01-05         2
1987-01-06         3
1987-01-07         4
1987-01-08         5