获取日期列表中的连续日期，且表示成*年*月*日~*年*月*日

获取日期列表中的连续日期，且表示成*年*月*日~*年*月*日

#coding:utf-8
#作者    ：思
#创建时间：2021/12/16 14:36 
#功能    ：获取日期列表中的连续日期，且表示成*年*月*日~*年*月*日
import numpy as np
from itertools import groupby
import datetime
dlist = ['2021-03-01', '2021-04-01', '2021-06-06', '2021-06-25', '2021-06-26',
       '2021-06-27', '2021-07-28', '2021-08-29','2021-08-30','2021-08-31','2021-09-01']

def calculate_consecutive_dates(dlist):
    dates = [datetime.datetime.strptime(i, "%Y-%m-%d").date() for i in dlist]
    date_ints = [d.toordinal() for d in dates]
    print(date_ints)

    reslist = []
    fun = lambda x: x[1]-x[0]
    for k, g in groupby(enumerate(date_ints), fun):
        l1 = [j for i, j in g ]    # 连续数字的列表

        if len(l1)>2:#时间连续的点超过2天
            conl_start = datetime.date.fromordinal(l1[0])
            conl_end = datetime.date.fromordinal(l1[-1])
            conl_start = conl_start.strftime("%Y-%m-%d")
            conl_end = conl_end.strftime("%Y-%m-%d")
            conl_str = conl_start+'~'+conl_end
            print(conl_start,conl_end)
            print(conl_str)
            reslist.append(conl_str)
        else:
            for i in l1:
                i = datetime.date.fromordinal(i)
                i = i.strftime("%Y-%m-%d")
            reslist.append(i)
    print(reslist)
    # print('，'.join(reslist))
    # ['2021-03-01', '2021-04-01', '2021-06-06', '2021-06-25~2021-06-27', '2021-07-28', '2021-08-29~2021-09-01']


if __name__ == '__main__':
    calculate_consecutive_dates(dlist)