python对日期区间进行拆分，每3个月为1个小区间

对于日期区间进行拆分，每3个月为1个小区间，具体有如下要求：

代码如下：

import pandas as pd
import numpy as np
import datetime

def gen_pd_date_range(start_date, end_date):
    s = datetime.datetime.strptime(start_date,"%Y-%m-%d")
    e = datetime.datetime.strptime(end_date,"%Y-%m-%d")
    delta = (e - s).days
    if delta<=90:
        return np.transpose([[start_date],[end_date]])

    #如果区间大于90天 进行拆分
    end_labels = [x[:10] for x in pd.date_range(start_date, end_date,
                                    freq='M').astype(np.str_)][2::3]

    start_labels = [start_date] + [str(x+pd.DateOffset(days=1))[:10] for x in
                    pd.date_range(start_date, end_date, freq='M')][2::3]

    if end_labels[-1] == end_date:
        return np.transpose([start_labels[:-1], end_labels])
    else:
        return np.transpose([start_labels, end_labels + [end_date]])

start_date, end_date = '2019-02-01', '2020-04-30'
df = pd.DataFrame(gen_pd_date_range(start_date, end_date),columns=['begindate','enddate'])
print(df)

输出结果如下：