在Scala Spark中将数据帧拆分为多个数据帧

import org.apache.spark.sql.SparkSession

object DataFrameSplit {
  def main(args: Array[String]): Unit = {
    val spark = SparkSession.builder()
      .appName("DataFrameSplit")
      .master("local")
      .getOrCreate()

    // 创建一个数据帧
    val data = Seq(
      ("Alice", 25),
      ("Bob", 30),
      ("Charlie", 35),
      ("David", 40),
      ("Eva", 45)
    )
    val df = spark.createDataFrame(data).toDF("Name", "Age")

    // 将数据帧拆分为两个数据帧，比例为0.7和0.3
    val Array(df1, df2) = df.randomSplit(Array(0.7, 0.3))

    // 打印拆分后的数据帧
    df1.show()
    df2.show()

    spark.stop()
  }
}