Pyspark判断日期是否为休息日并做成日历表udf

for your wish

已于 2024-09-20 10:56:37 修改

阅读量303

点赞数

文章标签： python 开发语言

于 2023-04-20 11:14:48 首次发布

Hive 同时被 2 个专栏收录

55 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

Python

33 篇文章

订阅专栏

使用PyCharm安装chinesecalender库来判断日期是否为休息日，并结合另一篇博文关于Hive日历表的知识，创建基础日历表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pycharm安装chinesecalender ，file-setting-python interpreter加号找chinesecalender 然后安装

from pyspark.sql import SparkSession
from pyspark.sql.types import IntegerType
import sys
from datetime import datetime
from chinese_calendar import is_workday

def judge_workday(a):
  # str转date类型  如果是date类型可直接传入
  a=datetime.strptime(a, '%Y-%m-%d').date()
  duration = 0  # 初始化  工作日
  if is_workday(a):
    duration=1
  else:
    duration=0
  return duration

# date = datetime.datetime.now().date()
# date='2023-04-23'
# print(judge_workday(date))

if __name__ == "__main__":
    # .master("local") \
    spark = SparkSession.builder \
        .appName("makeCalender") \
        .enableHiveSupport() \
        .getOrCreate()

    spark.udf.register("udf_judge_workday", jud