根据holiday和workingday生成新的特征type,区分是节假日、工作日还是周末
时间: 2024-02-12 08:05:44 浏览: 121
chinese-workday:中国工作日,用于查询某天是否是工作日、节假日、是否要补班
你可以在上面的代码基础上,添加一个新的函数来生成新的特征type,用于区分是节假日、工作日还是周末。下面是一个示例代码:
```python
def generate_type(date):
if is_holiday(date):
return '节假日'
elif is_weekend(date):
return '周末'
else:
return '工作日'
# 测试代码
date1 = datetime.date(2022, 2, 12)
date2 = datetime.date(2022, 2, 14)
date3 = datetime.date(2022, 2, 15)
print(generate_type(date1)) # 节假日
print(generate_type(date2)) # 周末
print(generate_type(date3)) # 工作日
```
在这个示例中,generate_type()函数接受一个日期作为参数,返回这个日期的类型。如果这个日期是节假日,返回"节假日";如果这个日期是周末,返回"周末";否则返回"工作日"。
阅读全文