set efficently specific rows in a sparse matrix to 0

Question

Marko 2021-9-3

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1445994-set-efficently-specific-rows-in-a-sparse-matrix-to-0

评论： Walter Roberson 2021-9-3

采纳的回答： Walter Roberson

在 MATLAB Online 中打开

Hello comunity,

I have a sparse matrix J and i want to set specific rows to 0.

My code's like this:

J = sparse(blkdiag(rand(1000,1000),rand(100,100),rand(200,200))); % just for example
vec = [1:10,20:400,800:1100]';   % just for example
J(vec,:) = 0;

Is there a faster way how to set the specific lines of J to 0?

best regards,

Marko

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Walter Roberson 2021-9-3

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1445994-set-efficently-specific-rows-in-a-sparse-matrix-to-0#answer_780119

在 MATLAB Online 中打开

Would it be practical in your situation to transpose the matrix, so that you were setting columns to 0 instead of rows? With the implementation of sparse(), it is more efficient to change multiple items in one column.

Beyond that: the usual efficiency hint is to find() the non-zero objects of interest, and then use those indices to assign new values.

It would be interesting to benchmark,

J(vec,:) = 0;

compared to

[r,c] = find(J(vec,:));
J(sub2ind(size(J), vec(r), c)) = 0;

compared to

[r,c,s] = find(J(vec,:));
J = J - sparse(vec(r), c, s, size(J,1), size(J,2));

The third of those is locating the non-zero elements of those rows of J, along with the values there, and building a new sparse array that contains only those elements, and then subtracts the two.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

Answer 2

Marko 2021-9-3

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1445994-set-efficently-specific-rows-in-a-sparse-matrix-to-0#answer_780179

在 MATLAB Online 中打开

here are the benchmark results:

    ver    Var2 
    ___    _____
     1     2.242
     2       NaN
     3      0.87

for version 2 i get the error:

Error using sub2ind (line 51)
The subscript vectors must all be of the same size.
Error in setBC (line 12)
J(sub2ind(size(L), vec(r), c)) = 0;

so your code snippet (version 3) saved me 61% or 1,3s every time i ran a computation.

Thank you very much

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Walter Roberson 2021-9-3

In your posted question, vec is a column vector. And the code I posted works when vec is a column vector. But if vec were a row vector instead then you would get the error you mention.

请先登录，再进行评论。