python生成时间序列_python生成时间序列(date_range)

importreimportcalendarimportdatetimeclassFormatError(ValueError):pass

classDate(object):

@classmethoddef date_range(cls, start=None, end=None, periods=None, freq=None, input_format=None, out_format=None):"""生成时间序列

:param start: 序列开始时间

:param end: 序列结束时间, 给定start时, 结束时间不包含end

:param periods: int, 生成的时间序列长度

:param freq: 要生成时间序列的时间间隔

:param out_format: 是否输出格式化后的字符串, 若要输出可指定输出格式. "%Y-%m-%d %H:%M:%S"

:param input_format: 若start或end是字符串且无法自动推断时间格式则需指定格式

:return: [date or date_str]"""start=cls.str_to_date(start, input_format)

end=cls.str_to_date(end, input_format)

out=[]if start is None and end andperiods:for i in range(periods-1):

old, end= cls.date_replace(end, cls._freq(freq), mod="-")if i ==0:

out.append(old)

out= [end] +outelif end is None and start andperiods:for i in range(periods-1):

old, start= cls.date_replace(start, cls._freq(freq), mod="+")if i ==0:

out.append(old)

out.append(start)elif periods is None and start andend:

i=0whileTrue:

old, start= cls.date_replace(start, cls._freq(freq), mod="+")if i ==0:

out.append(old)

i+= 1

if start

out.append(start)else:break

else:raise ValueError("start, end, periods 须且只能指定其中两个")if out_format isTrue:

out= [str(i) for i inout]elif out_format is notNone:

out= [i.strftime(out_format) for i inout]returnout

@staticmethoddef date_replace(date, freq=(0, )*6, mod="+"):

timedelta= datetime.timedelta(days=freq[2], hours=freq[3], minutes=freq[4], seconds=freq[5])if mod == "+":if sum(freq[:2]) ==0:

old=date

date= date +timedeltaelif sum(freq[2:]) ==0:

y= date.year + freq[0] + (date.month + freq[1] - 1) // 12m= (date.month + freq[1] - 1) % 12 + 1old= date.replace(day=calendar.monthrange(date.year, date.month)[1])

date= date.replace(year=y, month=m, day=calendar.monthrange(y, m)[1])else:raise ValueError("'年月' 不能同时和 '日时分秒' 作为间隔")elif mod == "-":if sum(freq[:2]) ==0:

old=date

date= date -timedeltaelif sum(freq[2:]) ==0:

y= date.year - freq[0] + (date.month - freq[1] - 1) // 12m= (date.month - freq[1] - 1) % 12 + 1old= date.replace(day=calendar.monthrange(date.year, date.month)[1])

date= date.replace(year=y, month=m, day=calendar.monthrange(y, m)[1])else:raise ValueError("'年月' 不能同时和 '日时分秒' 作为间隔")else:raise ValueError("mod值只能是 '+' 或 '-'")returnold, date

@staticmethoddef _freq(freq=None):"""设置时间间隔

:param freq: "Y2m3d4H5M6S" 表示间隔 1年2月3日4时5分6秒

:return: [年, 月, 日, 时, 分, 秒]"""freq_= [0] * 6

if freq isNone:

freq_[2] = 1

returnfreq_for n, i in enumerate(["Y", "m", "d", "H", "M", "S"]):

r= f'((\d*){i})'s=re.search(r, freq)ifs:

freq_[n]= int(s.group(2)) if s.group(2) else 1

returnfreq_

@staticmethoddef str_to_date(string, format_=None):"""字符串转时间, 默认自动推断格式

:param string: 时间字符串

:param format_: 格式

:return: 对应的时间类型, 输入非字符串则原值输出"""

if notisinstance(string, str):returnstringifformat_:returndatetime.datetime.strptime(string, format_)

s= re.match(r'(\d{4})\D+(\d{1,2})\D+(\d{1,2})(?:\D+(\d{1,2}))?(?:\D+(\d{1,2}))?(?:\D+(\d{1,2}))?\D*$', string)ifs:

result= [int(i) for i in s.groups() ifi]return datetime.datetime(*result)

s= re.match(r'(\d{4})\D*(\d{2})\D*(\d{2})\D*(\d{2})?\D*(\d{2})?\D*(\d{2})?\D*$', string)ifs:

result= [int(i) for i in s.groups() ifi]return datetime.datetime(*result)else:raise FormatError("自动推断失败, 请指定format_")