使用微风计算稀疏矩阵中的非零元素

Question

我是微风的新手。我似乎无法计算 CSCMatrix 中每行的非零元素数。

让我们看下面的例子：

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

这是一个维度为 3x2 的稀疏矩阵：

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

我想要的是计算每行非零元素的数量，因此输出应该是以下形式的向量：

|0|
|1|
|1|

用 numpy 很容易做到，但我似乎无法用微风做到。