scala - 在 Scalala 中切片 SparseVector 以获得 SparseVector

Question

我想在scalala中切片一个稀疏向量并得到另一个稀疏向量。法线切片总是给出标准向量。我可以在切片时提供一个隐式函数来执行此操作吗？如果可以，它会是什么样子？

例如，我想完成以下任务：

val v:SparseVectorRow[Double] = ...
val v2:SparseVectorRow[Double] = v(List(1,3,5))

最后，我使用的是 Scalala 版本 1.0.0.RC3-SNAPSHOT。

提前致谢。

Answer 1

是的，您可以定义自己的适当参数化的CanSliceVector. 以下不是通用的或不一定有效的，但它有效：

import scalala.generic.collection.CanSliceVector

implicit object svrSlicer extends
  CanSliceVector[SparseVectorRow[Double], Int, SparseVectorRow[Double]] {
  def apply(from: SparseVectorRow[Double], keys: Seq[Int]) =
    SparseVector.create(from.size)(keys.map(k => (k, from(k))): _*).t
}

val v: SparseVectorRow[Double] = SparseVector.create(10)((0, 1.0), (3, 2.0)).t
val v2: SparseVectorRow[Double] = v(List(1, 3, 5))

您可以在 Scalala 源代码中四处寻找以找出执行此操作的“正确”方法。