如何在spark scala中找到数据帧中的词组计数？

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

// 创建SparkSession
val spark = SparkSession.builder()
  .appName("Word Count")
  .getOrCreate()

// 读取数据帧
val df = spark.read.text("path/to/input/file.txt")

// 定义一个UDF来拆分文本列为单词
val splitWords = udf((text: String) => text.split("\\s+"))

// 拆分文本列为单词，并将每个单词映射为键值对
val wordCounts = df.withColumn("words", splitWords(col("text")))
  .select(explode(col("words")).as("word"))
  .groupBy("word")
  .count()

// 打印词组计数结果
wordCounts.show()

// 推荐的腾讯云相关产品和产品介绍链接地址：
// - 腾讯云计算服务：https://cloud.tencent.com/product/cvm
// - 腾讯云数据库：https://cloud.tencent.com/product/cdb
// - 腾讯云人工智能：https://cloud.tencent.com/product/ai
// - 腾讯云物联网：https://cloud.tencent.com/product/iotexplorer
// - 腾讯云移动开发：https://cloud.tencent.com/product/mobdev
// - 腾讯云存储：https://cloud.tencent.com/product/cos
// - 腾讯云区块链：https://cloud.tencent.com/product/baas
// - 腾讯云元宇宙：https://cloud.tencent.com/product/vr