Изменения ненулевых в некоторых столбцах разреженной матрицы

Question

У меня есть разреженная матрица в MATLAB:

N=1000; 
P=0.01; 
A=sprand(N,N,P); 

и я хочу, чтобы изменить все не нулевые элементы в определенных столбцах в них.

То есть, что-то вроде этого:

c=randi(N,[1,round(N/10)]); 
A(non zeros at columns c)=1; 

Конечно, это можно сделать в цикле, но это явно не решение, которое я ищу. Я пробовал несколько решений с использованием nnz, nonzeros, spfun - но без каких-либо проблем. Может ли кто-нибудь придумать простой способ сделать это?

Спасибо, ELAD

Answer 1

Вы можете сделать это следующим образом:

A(:,c) = abs(sign(A(:,c))); % take the absolute value of the sign for all entries 
          % in the submatrix defined by the columns in c, and 
          % then assign the result back 

Эквивалентное

A(:,c) = logical(A(:,c); 

или

A(:,c) = A(:,c)~=0; 

Это может быть не так быстро, потому что они обрабатывают все записи в этих столбцах, а не только ненулевые записи. Dohyun's approach, вероятно, быстрее.

Answer 2

Вы можете попробовать это

N = 1000; 
P = 0.01; 
A = sprand(N,N,P); 

c = unique(randi(N,[1,round(N/10)]))'; % sorted column index 
[r,cind] = find(A(:,c)); 

A(sub2ind([N,N],r,c(cind)))=1; 

Answer 3

относящиеся к Luis Mendos answer, но немного проще

A(:,c) = ceil(A(:,c));