Scala:相当于 np.digitize 来存储数据

Question

对于scala，我有一些数据，例如。

val values = Seq(0, 2, 10, 50)

我定义了桶，例如。

val buckets = TreeMap[Int, Double]((0, -0.001),
                                   (1, 1.5),
                                   (2, 5),
                                   (3, 20)

然后，我希望从我的值中拥有存储桶索引，例如。

val result = Seq(0, 1, 2, 3)

在python中，

这可以用np.digitize来完成，在scala中，我找不到Nd4j或Breeze的等价物。

有没有优化的解决方案？

Answer 1

也许您使用的是较旧的 Breeze 版本。如果包括：

libraryDependencies += "org.scalanlp" %% "breeze" % "0.13.2"

在 .sbt 文件中，您可以使用该功能。例如：

import breeze.stats._
val arr1 = Array(-3, 0.5, 1, 1.5, 4)
val arr2 = Array(0, 1, 2)
digitize(arr1, arr2)

它给

Array(0, 1, 1, 2, 3)