如何在apache spark中将数字列划分为范围并为每个范围分配标签？

import org.apache.spark.ml.feature.Bucketizer
import org.apache.spark.sql.functions._

val data = Seq(0.1, 2.5, 4.3, 6.7, 8.9, 10.2, 12.5, 14.8, 16.3, 18.9).toDF("numbers")

val splits = Array(Double.NegativeInfinity, 5.0, 10.0, Double.PositiveInfinity)

val bucketizer = new Bucketizer()
  .setInputCol("numbers")
  .setOutputCol("bucketedNumbers")
  .setSplits(splits)

val bucketedData = bucketizer.transform(data)

bucketedData.show()

+-------+---------------+
|numbers|bucketedNumbers|
+-------+---------------+
|    0.1|            0.0|
|    2.5|            0.0|
|    4.3|            0.0|
|    6.7|            1.0|
|    8.9|            1.0|
|   10.2|            2.0|
|   12.5|            2.0|
|   14.8|            2.0|
|   16.3|            2.0|
|   18.9|            2.0|
+-------+---------------+