从单个列创建多个列- Scala spark

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

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Create Multiple Columns")
  .master("local")
  .getOrCreate()

// 创建示例数据
val data = Seq(
  (1, "John", 25),
  (2, "Jane", 30),
  (3, "Bob", 35)
)

// 将数据转换为DataFrame
val df = spark.createDataFrame(data).toDF("id", "name", "age")

// 从单个列创建多个列
val dfWithNewColumns = df.select(
  col("id"),
  col("name"),
  col("age"),
  col("age").plus(1).as("age_plus_1"),
  concat(col("name"), lit(" is "), col("age")).as("name_age_concat")
)

// 显示结果
dfWithNewColumns.show()