用最多两列替换pyspark中的日期值

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, lit, when
from pyspark.sql.types import DateType

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

# 创建示例数据
data = [("2022-01-01", "2022-02-01"),
        ("2022-03-01", "2022-04-01"),
        ("2022-05-01", "2022-06-01")]

df = spark.createDataFrame(data, ["date1", "date2"])

# 替换日期值
df = df.withColumn("date1", when(col("date1") == lit("2022-01-01"), lit("2022-07-01")).otherwise(col("date1")))
df = df.withColumn("date2", when(col("date2") == lit("2022-02-01"), lit("2022-08-01")).otherwise(col("date2")))

# 显示结果
df.show()