在scala中用相同的条件替换数据帧中的两个不同的列值，复杂度最低

import org.apache.spark.sql.{DataFrame, SparkSession}
import org.apache.spark.sql.functions._

val spark = SparkSession.builder()
  .appName("Replace values in DataFrame columns")
  .master("local")
  .getOrCreate()

val df = spark.createDataFrame(Seq(
  ("A", "X", 10),
  ("B", "Y", 20),
  ("C", "Z", 30)
)).toDF("col1", "col2", "col3")

val condition = col("col1") === lit("A") && col("col2") === lit("X")
val newValue = lit("NEW_VALUE")

val replacedDf = df.withColumn("col1", when(condition, newValue).otherwise(col("col1")))
  .withColumn("col2", when(condition, newValue).otherwise(col("col2")))