如何更新Scala Dataframe的单个单元？

import org.apache.spark.sql.functions._

// 创建一个示例Dataframe
val df = spark.createDataFrame(Seq(
  (1, "John", 25),
  (2, "Amy", 30),
  (3, "David", 35)
)).toDF("id", "name", "age")

// 定义更新的条件
val condition = col("id") === 2

// 定义新值
val newValue = lit("Tom")

// 更新Dataframe
val updatedDF = df.withColumn("name", when(condition, newValue).otherwise(col("name")))

// 显示更新后的Dataframe
updatedDF.show()