在行中插入值(Spark - Scala)

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

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Insert value in row - Spark Scala")
  .master("local")
  .getOrCreate()

// 创建一个DataFrame
val data = Seq(
  Row(1, "John", 25),
  Row(2, "Jane", 30),
  Row(3, "Tom", 35)
)

val schema = new StructType()
  .add("id", IntegerType)
  .add("name", StringType)
  .add("age", IntegerType)

val df = spark.createDataFrame(spark.sparkContext.parallelize(data), schema)

// 在行中插入新的值
val newRow = Row(4, "Alice", 28)
val updatedDF = df.union(spark.createDataFrame(spark.sparkContext.parallelize(Seq(newRow)), schema))

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