python - 關于計算日期的問題

問題描述

先上代碼

def get_date(today,num): try:myday = datetime.datetime( int(today[0:4]),int(today[5:7]),int(today[8:10]) ) + datetime.timedelta(days=num)dt = myday.strftime(’%Y-%m-%d’) except Exception as e:print(e)print(’日期格式不合法’)return return dtif __name__==’__main__’: print(’計算日期列表...’) date_list=[] today=’2017-06-20’ for i in range(-90,0):date=get_date(today,i)ISOTIMEFORMAT = ’%Y-%m-%d’tm=time.strptime(date, ISOTIMEFORMAT)if time.asctime(tm)[0:3]!=’Sat’ and time.asctime(tm)[0:3]!=’Sun’: date_list.append(date)#去除雙休日else: print date,’is holiday,removed..’ #date_list就是日期列表

計算的是2017-6-20之前90天除了雙休日之外的所有日期，有誰有更簡單暴力的方法嗎？第三方庫推薦下也行。

問題解答

from datetime import datetime, timedeltadef get_date(today): date_format = ’%Y-%m-%d’ dt = datetime.strptime(today, date_format) for i in range(1, 91):dt = dt + timedelta(days=-1)if dt.weekday() > 0 and dt.weekday() < 6: yield dt.strftime(date_format)print list(get_date(’2017-06-20’))