使用PySpark数据框中的2列作为查找，以替换所有其他列

from pyspark.sql import SparkSession

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

# 创建示例数据框
data = [("Alice", 25, "Female"), ("Bob", 30, "Male"), ("Charlie", 35, "Male")]
df = spark.createDataFrame(data, ["Name", "Age", "Gender"])

# 使用两列作为查找条件，替换所有其他列的值
df = df.withColumn("Name", when(col("Age") > 30, "John").otherwise(col("Name"))) \
       .withColumn("Age", when(col("Age") > 30, 40).otherwise(col("Age"))) \
       .withColumn("Gender", when(col("Age") > 30, "Other").otherwise(col("Gender")))

# 打印替换后的数据框
df.show()