如果月份在整数列表中，则过滤Dataset<Row>

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

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

    // 假设有一个包含日期和其他列的数据集
    val data = Seq(
      ("2022-01-01", "data1"),
      ("2022-02-01", "data2"),
      ("2022-03-01", "data3"),
      ("2022-04-01", "data4")
    )

    import spark.implicits._

    // 将数据集转换为DataFrame
    val df = data.toDF("date", "value")

    // 添加一个名为month的列，提取日期中的月份
    val dfWithMonth = df.withColumn("month", month($"date"))

    // 定义一个整数列表，包含要筛选的月份
    val targetMonths = List(2, 3)

    // 过滤出包含指定月份的数据集
    val filteredDF = dfWithMonth.filter($"month".isin(targetMonths: _*))

    filteredDF.show()
  }
}