How to vectorize for loops

Hello.Here's my code.I want to improve it's efficiency. Therefore,I try to vectorize this for loop in this code.Hope anyone who mastering Matlab can help me solve this problem.Thank you all.
K = 2;
M = 3;
b = zeros(K,1);
A = [1,1,1;
1,1,2;
1,2,1;
1,2,2;
2,1,1;
2,1,2;
2,2,1;
2,2,2];
C = num2cell(A);
for i = 1:M
for j = 1:K^M
for q = 1:K % 1 --> [1;0;...;0] 2 --> [0;1;0;0...;0] K --> [0;0;0;...;1]
if C{j,i} == q
b(q,1) = 1;
C(j,i) = {b};
end
end
b = zeros(K,1);
end
end

 Accepted Answer

K = 2;
S = size(A);
[R,C] = ndgrid(1:S(1),1:S(2));
Z = zeros([K,S]);
Z(sub2ind([K,S],A,R,C)) = 1;
out = permute(num2cell(Z,1),[2,3,1]);

More Answers (0)

Categories

Find more on Loops and Conditional Statements in Help Center and File Exchange

Asked:

on 7 Jun 2018

Answered:

on 7 Jun 2018

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!