我想将dataframe中的一列月份数字更改为月份名称(pyspark)

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, when

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 创建示例DataFrame
data = [(1, 1), (2, 2), (3, 3)]
df = spark.createDataFrame(data, ["id", "month"])

# 定义一个UDF函数，将月份数字转换为月份名称
def get_month_name(month):
    month_names = ["January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"]
    return month_names[month - 1]

# 使用withColumn和when函数进行列转换
df = df.withColumn("month_name", when(col("month").isNull(), None).otherwise(get_month_name(col("month"))))

# 显示转换后的结果
df.show()