如何在PySpark中编写条件正则表达式替换？

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

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

# 创建示例数据
data = [("John Doe", 25), ("Jane Smith", 30), ("Bob Johnson", 35)]
df = spark.createDataFrame(data, ["name", "age"])

# 使用regexp_replace函数进行条件正则表达式替换
df = df.withColumn("name", regexp_replace(df.name, "John", "Tom"))

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