相当于 MATLAB spones 的 numpy

Question

这是 MATLAB 内置函数 spones(S) 的文档字符串:

spones

Replace nonzero sparse matrix elements with ones. R = spones(S) generates a matrix with the same sparsity structure as S, but with ones in the nonzero positions.

我想使用 numpy/scipy 数据结构(例如来自 scipy.sparse 的稀疏矩阵)来获得与此函数非常接近的等效项。我怎样才能有效地做到这一点？

Answer 1

x = ... some sparse matrix ...
y = x.copy().tocsr()
y.data.fill(1)