Creating a matrix of ones and zeros with location of ones based on array value
조회 수: 34 (최근 30일)
이전 댓글 표시
I need to acheive the following without a loop (for speed):
I have a vertical array where the value of each row tells me which column in a matrix of zeros should be replaced with a one.
For example, if I input:
y = [2; 3];
Then I want MATLAB to produce output:
Y = [0,1,0; 0, 0, 1];
I can acheive the desired output with the following for loop:
Y = zeros(2,3);
for j = 1:3
Y(:,j) = (y==j);
end
In reality, my matrices are much larger and I want to avoid using for loops for speed. Is there anyway I can do this without a for loop?
댓글 수: 1
madhan ravi
2019년 4월 9일
편집: madhan ravi
2019년 4월 9일
@ Melissa Moore :Are you interested in a faster solution still? If yes another solution can be proposed.
채택된 답변
Kevin Rawson
2019년 4월 8일
One possible way is:
Y = zeros(length(y), max(y));
Y(sub2ind(size(Y), (1:length(y))', y)) = 1;
댓글 수: 0
추가 답변 (2개)
A. Sawas
2019년 4월 8일
편집: A. Sawas
2019년 4월 8일
The best solution I can think of is to use a for loop but will be faster than the one you showed because it will do one scalar assignment in each iteration compared to a vector assignment.
for j=1:2
Y(j,y(j)) = 1;
end
댓글 수: 1
Kevin Rawson
2019년 4월 8일
편집: Kevin Rawson
2019년 4월 8일
A. Sawas, your method is faster for small arrays, but not for large arrays. I'm using your 10,000x10,000 example with y assigned to the diagonal (see code above).
Sawas method (run 10000 times):
tic;
for i=1:10000
for j=1:length(y)
Y(j,y(j)) = 1;
end
end;
toc;
%Elapsed time is 1.682664 seconds.
Rawson method (run 10000 times):
tic;
for i=1:10000
Y(ind2sub(1:length(y), y)) = 1;
end
toc;
%Elapsed time is 0.596671 seconds.
Melissa Moore
2019년 4월 8일
댓글 수: 3
Kevin Phung
2019년 4월 8일
Is Melissa's solution a valid solution?... IMO, I is a diagonal matrix where there are already existing 1's. And this is not what you want.
A. Sawas's is valid and more efficient.
Kevin Rawson
2019년 4월 8일
편집: Kevin Rawson
2019년 4월 9일
However, as A. Sawas pointed out, this is an extremely slow operation, ~700 times slower than A. Sawas' method, and ~1988 times slower than my method for a 10,000x10,000 array.
참고 항목
카테고리
Help Center 및 File Exchange에서 Logical에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!