spark:如何将行合并到json数组中

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

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Merge Rows into JSON Array")
  .getOrCreate()

// 创建一个示例数据集
val data = Seq(
  ("Alice", "Apple"),
  ("Bob", "Banana"),
  ("Alice", "Orange"),
  ("Bob", "Grape")
).toDF("Name", "Fruit")

// 使用groupBy和collect_list函数将行合并到JSON数组中
val result = data.groupBy("Name")
  .agg(collect_list("Fruit").as("Fruits"))
  .toJSON

// 显示结果
result.show(false)