在scala函数中传递RDD。输出数据帧

import org.apache.spark.sql.{DataFrame, SparkSession}
import org.apache.spark.rdd.RDD

val spark = SparkSession.builder()
  .appName("RDD to DataFrame")
  .master("local")
  .getOrCreate()

def rddToDataFrame(rdd: RDD[(String, Int)]): DataFrame = {
  // 将RDD转换为数据帧
  val df = spark.createDataFrame(rdd)
    .toDF("name", "age") // 自定义列名，根据实际情况修改

  // 返回数据帧
  df
}

def main(args: Array[String]): Unit = {
  // 创建一个RDD
  val rdd = spark.sparkContext.parallelize(Seq(("John", 25), ("Alice", 30), ("Bob", 35)))

  // 调用函数将RDD转换为数据帧
  val df = rddToDataFrame(rdd)

  // 打印数据帧内容
  df.show()
}