使用 breeze 计算稀疏矩阵中的 non-zero 个元素

Counting non-zero elements in a sparse matrix using breeze

我是 breeze 的新手。我似乎无法计算 CSCMatrix.

中每行的 non-zero 元素的数量

让我们举个例子:

val matrix = CSCMatrix.tabulate(3, 2)((x,y) => x*y)

这是一个 3x2 维的稀疏矩阵:

|0|0|
|0|1|
|0|2|

我想要的是计算每行 non-zero 个元素的数量,因此输出应该是以下形式的向量:

|0|
|1|
|1|

用 numpy 很容易做到,但我似乎无法用 breeze 做到。

如果有人仍然需要这个,您可以按以下方式使用矩阵向量乘法来完成:

matrix * DenseVector.ones[Int](matrix.cols)