python - Pandas - 从 DateTimeIndex 创建列

Question

我有一个带有日期时间索引的 pandas DataFrame。我想创建一个由我的数据的年份和月份组成的 label 列。

我发现的方法是复制该列并对其进行操作(我必须这样做，因为 DateTimeIndex 没有 apply 方法).我确信一定有一种方法可以直接对索引进行操作，但我找不到它:

import pandas  as pd
import numpy as np


df = pd.DataFrame(index=pd.date_range(start="2012-01-01", end="2013-01-01", freq='D'), data=range(367))

monthly = df.resample("M")
monthly["label"] = monthly.index
monthly["label"] = monthly["label"].apply(lambda x: x.strftime("%Y-%m"))

monthly.head() 给我的地方:

                0    label
2012-01-31   15.0  2012-01
2012-02-29   45.0  2012-02
2012-03-31   75.0  2012-03
2012-04-30  105.5  2012-04
2012-05-31  136.0  2012-05

这正是我想要的，我只想在没有我用作解决方法的源代码中的倒数第二行的情况下执行此操作。

Answer 1

可能有一种更灵活的方法可以直接获取 strftime 结果，但通常当您想要将索引视为一列而不实际将其作为一个列时，您可以调用 .to_series()相反:

>>> monthly.index.to_series().head()
2012-01-31   2012-01-31
2012-02-29   2012-02-29
2012-03-31   2012-03-31
2012-04-30   2012-04-30
2012-05-31   2012-05-31
Freq: M, dtype: datetime64[ns]
>>> monthly.index.to_series().apply(lambda x: x.strftime("%Y-%m")).head()
2012-01-31    2012-01
2012-02-29    2012-02
2012-03-31    2012-03
2012-04-30    2012-04
2012-05-31    2012-05
Freq: M, dtype: object