python - 将 pandas 多索引折叠为单个索引

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