如何使用python或pyspark获取包含年份(此格式为202114)的周数列表

import datetime

def get_week_numbers(year):
    start_date = datetime.datetime(year, 1, 1)
    end_date = datetime.datetime(year, 12, 31)
    num_weeks = (end_date - start_date).days // 7 + 1
    week_numbers = [start_date + datetime.timedelta(weeks=i) for i in range(num_weeks)]
    return [week_number.strftime("%Y%W") for week_number in week_numbers]

year = 2021
week_numbers = get_week_numbers(year)
print(week_numbers)

from pyspark.sql import SparkSession
from pyspark.sql.functions import udf
from pyspark.sql.types import StringType
import datetime

spark = SparkSession.builder.getOrCreate()

def get_week_numbers(year):
    start_date = datetime.datetime(year, 1, 1)
    end_date = datetime.datetime(year, 12, 31)
    num_weeks = (end_date - start_date).days // 7 + 1
    week_numbers = [start_date + datetime.timedelta(weeks=i) for i in range(num_weeks)]
    return [week_number.strftime("%Y%W") for week_number in week_numbers]

get_week_numbers_udf = udf(get_week_numbers, StringType())

year = 2021
df = spark.range(1).select(get_week_numbers_udf(year).alias("week_numbers"))
df.show(truncate=False)