我是否可以使用regexp_replace或其他等效工具用一行代码替换一个pyspark dataframe列中的多个值？

from pyspark.sql import SparkSession
from pyspark.sql.functions import regexp_replace

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

# 创建示例DataFrame
data = [("John", "A"), ("Mike", "B"), ("Anna", "C"), ("Tom", "A")]
df = spark.createDataFrame(data, ["Name", "Grade"])

# 定义要替换的值和替换后的值
replace_dict = {"A": "Excellent", "B": "Good", "C": "Average"}

# 使用regexp_replace函数替换列中的值
df = df.withColumn("Grade", regexp_replace("Grade", "|".join(replace_dict.keys()), 
                                           lambda x: replace_dict[x]))

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