numpy相当于MATLAB spones

时间:2012-05-14 10:31:01

标签: numpy python-3.x scipy

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

  

spones

     

用1替换非零稀疏矩阵元素。       R = spones(S)生成具有相同稀疏度的矩阵       结构为S,但是非零位置。

我希望使用numpy / scipy数据结构(例如来自scipy.sparse的稀疏矩阵)与此函数等效。我怎样才能有效地做到这一点?

1 个答案:

答案 0 :(得分:6)

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