在spark scala中，为数据帧中的每个组采样不同数量的随机行

import org.apache.spark.sql.{SparkSession, functions}

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

// 导入隐式转换
import spark.implicits._

// 创建示例数据帧
val data = Seq(
  ("group1", "row1"),
  ("group1", "row2"),
  ("group1", "row3"),
  ("group2", "row4"),
  ("group2", "row5"),
  ("group2", "row6"),
  ("group2", "row7")
).toDF("group", "row")

// 定义每个组的采样比例
val sampleRatio = Map(
  "group1" -> 0.5,
  "group2" -> 0.3
)

// 使用sampleBy函数进行采样
val sampledData = data.sampleBy("group", sampleRatio, seed = 42)

// 显示采样结果
sampledData.show()