Scala/Spark :如何检查数据帧是否包含特定的列列表？

import org.apache.spark.sql.SparkSession

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

    // 创建示例数据帧
    val data = Seq(
      ("Alice", 25, "Female"),
      ("Bob", 30, "Male"),
      ("Charlie", 35, "Male")
    )
    val columns = Seq("Name", "Age", "Gender")
    val df = spark.createDataFrame(data).toDF(columns: _*)

    // 检查数据帧是否包含特定的列列表
    val requiredColumns = Seq("Name", "Age")
    val containsAllColumns = requiredColumns.forall(df.columns.contains)

    // 输出结果
    if (containsAllColumns) {
      println("数据帧包含所有指定的列列表")
    } else {
      println("数据帧不包含所有指定的列列表")
    }

    // 停止SparkSession
    spark.stop()
  }
}