如何从Spark中的id列表中找到包含id的所有数据？

// 导入Spark相关的库
import org.apache.spark.sql.SparkSession

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Filter Example")
  .getOrCreate()

// 创建一个包含id和data的DataFrame
val data = Seq(
  (1, "data1"),
  (2, "data2"),
  (3, "data3"),
  (4, "data4"),
  (5, "data5")
).toDF("id", "data")

// 定义要查找的id列表
val idList = List(2, 4)

// 使用filter操作从DataFrame中找到包含id的所有数据
val filteredData = data.filter(row => idList.contains(row.getAs[Int]("id")))

// 显示结果
filteredData.show()