如何在spark/scala中根据索引位置查找for each的值和

import org.apache.spark.sql.SparkSession

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

    val data = Seq("apple", "banana", "orange", "grape")
    val rdd = spark.sparkContext.parallelize(data)

    val indexedRdd = rdd.zipWithIndex()

    val indexToLookup = 2
    val value = indexedRdd.filter { case (element, index) => index == indexToLookup }
      .map { case (element, index) => element }
      .first()

    println(s"The value at index $indexToLookup is $value")

    spark.stop()
  }
}