Python计算两个日期间的工作日(交易日),去除周末节假日

首先区分一下这里的工作日和节假日是不同的。
工作日是指去掉节假日即周末,但是不去掉调休的日期。
交易日主要针对金融行业,是指去掉节假日和周末,因为周末金融市场也是停盘的,所以即使周末因为调休工作,也不算做交易日。
我们这里主要用到Python中datetime模块。还有一个特别好用的模块就是chinesecalendar,它主要能够实时更新中国的节假日有哪些,简直就是一个利器。
首先安装chinesecalendar也非常容易,我是在Anaconda Promot中安装的:

pip install chinesecalendar

安装完成之后,下面看一下两个方法的使用:

  • is_workday:判断是否为工作日,返回布尔值
  • is_holiday:判断是否为节假日,返回布尔值
from chinese_calendar import is_workday, is_holiday
import datetime
day1 = datetime.date(2020,1,18)
day2 = datetime.date(2020,1,19)
day3 = datetime.date(2020,1,31)
print(is_workday(day1)) # False
print(is_holiday(day1)) # True
print(is_workday(day2)) # True
print(is_workday(day3)) # False

2020-01-18是正常周日,所以不是工作日;2020-01-19是春节调休的工作日;2020-01-31是正月初七,是受今年新冠疫情影响延长的春节假期,也不算工作日。是不是很厉害,连这个都可以识别。有了它,我们就可以轻松识别我们的节假日了。
下面计算两个日期间的工作日天数:

def workdays(start,end):
    '''
    计算两个日期间的工作日
    start:开始时间
    end:结束时间
    '''
    from datetime import datetime,timedelta
    from chinese_calendar import is_workday
     # 字符串格式日期的处理
    if type(start) == str:
        start = datetime.strptime(start,'%Y-%m-%d').date()
    if type(end) == str:
        end = datetime.strptime(end,'%Y-%m-%d').date()
    # 开始日期大,颠倒开始日期和结束日期
    if start > end:
        start,end = end,start
    counts = 0
    while True:
        if start > end:
            break
        if is_workday(start):
            counts += 1
        start += timedelta(days=1)
    return counts

封装函数workdays(start,end)。可以接受形式为’%Y-%m-%d’字符串格式或日期格式的日期。下面运行看一下效果:

print(workdays('2020-01-19','2020-01-24'))
# 5
from datetime import date
print(workdays(date(2020,1,19),date(2020,1,24))) # 5

2020-01-19是调休工作日,所以一共是5天。
计算交易日天数:

def tradedays(start,end):
    '''
    计算两个日期间的工作日
    start:开始时间
    end:结束时间
    '''
    from datetime import datetime,timedelta
    from chinese_calendar import is_holiday
    # 字符串格式日期的处理
    if type(start) == str:
        start = datetime.strptime(start,'%Y-%m-%d').date()
    if type(end) == str:
        end = datetime.strptime(end,'%Y-%m-%d').date()
    # 开始日期大,颠倒开始日期和结束日期
    if start > end:
        start,end = end,start
        
    counts = 0
    while True:
        if start > end:
            break
        if is_holiday(start) or start.weekday()==5 or start.weekday()==6:
            start += timedelta(days=1)
            continue
        counts += 1
        start += timedelta(days=1)
    return counts

这里主要是需要排除所有的节假日和周六日。
运行看下效果:

print(tradedays('2020-01-19','2020-01-24')) # 4
from datetime import date
print(workdays(date(2020,1,24),date(2020,1,19))) # 4

因为2020-01-19虽然是工作日,但是是周日,所以不算交易日。
两个函数一个是计算工作日,一个计算交易日,都可以直接拿去使用,注意传参的格式,或者自行进行修改使用。

你可能感兴趣的:(Python计算两个日期间的工作日(交易日),去除周末节假日)