Scala：相当于numpy.where（）[0]

Question

在python中，如果x是一个numpy数组，我只需要写一下：

    np.where(x > 0)[0]

以获取否定条目的索引。 Scala中是否有相同的功能？

Answer 1

您可以使用findAll方法。一个例子：

val matrix = DenseVector.horzcat(DenseVector(0.0, -1.0), DenseVector(0.0, 0.0))
val idxs = matrix.findAll(_ < 0.0) // Vector((1, 0))

Answer 2

我最近使用的是如果x是一个数组，我可以使用：

val indexes = x.zipWithIndex.filter(_._1 > 0).map(_._2)