python - 将数据框转换为季度

Question

我有一个数据框(df)。日期变量是一个字符串。我想将其转换为日期并将其重新格式化为季度末的日期。示例如下:

df:
         Date
0     201601
1     201602
2     201603
3     201604

201601 是 2016 年第一季度，201604 是 2016 年第四季度。期望的结果是:

df:
          Date
0     2016-03-31
1     2016-06-30
2     2016-09-30
3     2016-12-31

我已经尝试过，但没有成功。

df['date'] = pd.to_datetime(df.Date, format = '%Y%q')

谢谢!

Answer 1

您可以定义一个函数来计算日期，然后应用该函数。

使用日历模块 ( https://docs.python.org/3.7/library/calendar.html#calendar.monthrange ) 中的 Monthrange 函数，您的函数可能类似于:

from calendar import monthrange
from datetime import datetime   

def end_quarter(quarter):
    year = int(quarter[:4])
    month = int(quarter[-2:]) * 3
    day = monthrange(year, month)[1]
    return datetime(year, month, day).date()

然后您可以使用以下方法应用它:

df["Date"] = df["Date"].apply(end_quarter)