Scala dataframe是否根据其他列添加新列？

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

val spark = SparkSession.builder()
  .appName("Adding New Column")
  .master("local")
  .getOrCreate()

// 创建一个示例 DataFrame
val data = Seq(("Alice", 25, "female"), ("Bob", 30, "male"), ("Charlie", 35, "male"))
val df: DataFrame = spark.createDataFrame(data).toDF("name", "age", "gender")

// 使用 withColumn 方法添加新列
val dfWithNewColumn = df.withColumn("ageGroup", when($"age" < 30, "young").otherwise("adult"))

// 打印结果
dfWithNewColumn.show()