使用 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)
我是 breeze 的新手。我似乎无法计算 CSCMatrix
.
让我们举个例子:
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)