Scala:如何获取数据帧中的一系列行

import org.apache.spark.sql.{DataFrame, Row}
import org.apache.spark.sql.functions.col

def getRows(dataFrame: DataFrame, rowIndices: Seq[Int]): DataFrame = {
  val rows = dataFrame.collect() // 将数据帧中的所有行收集到一个数组中
  val selectedRows = rowIndices.map(rows(_)) // 根据索引选择特定的行
  val selectedDataFrame = dataFrame.sparkSession.createDataFrame(selectedRows, dataFrame.schema) // 创建一个新的数据帧
  selectedDataFrame
}

val dataFrame: DataFrame = ... // 假设有一个名为dataFrame的数据帧
val rowIndices: Seq[Int] = Seq(0, 2, 4) // 假设要获取索引为0、2和4的行

val selectedDataFrame = getRows(dataFrame, rowIndices)