How can I change for-loop into matrix way?

조회 수: 1 (최근 30일)
icdi
icdi 2021년 6월 7일
댓글: icdi 2021년 6월 8일
I'd like to vectorize the for-loop below, but cannot come up with any idea. Could you help me?
for i=1:100
for j=1:1000
A(i+1,j)= B(find(C(B==A(i,j),:,D==E(i))>F(i,j),1));
end
end
  댓글 수: 9
icdi
icdi 2021년 6월 8일
편집: icdi 2021년 6월 8일
Yes all of your points are right.
D is 3x1. For all i and j, A(i,j) is supposed to have one of elements of B, and E(i) also has one of elements of D. So this code can give the series of numbers between 1 and 15.
Matt J
Matt J 2021년 6월 8일
편집: Matt J 2021년 6월 8일
So the elements of B and D are all unique? Also, what is supposed to happen if find() returns empty [], i.e., if
C(B==A(i,j),k,D==E(i)) < F(i,j)
for all k=1...15?

댓글을 달려면 로그인하십시오.

채택된 답변

Matt J
Matt J 2021년 6월 8일
This gives some speed-up.
%Fake Data
m=100;n=1000;
B=rand(15,1);
D=rand(3,1);
A=B(randi(15,m+1,n));
E=D(randi(3,m+1,1));
C=rand(15,15,3);
F=0.3*ones(size(A));
%Original version
tic;
for i=1:m
for j=1:n
A(i+1,j)= B(find(C(B==A(i,j),:,D==E(i))>F(i,j),1));
end
end
toc
Elapsed time is 0.372233 seconds.
A1=A;
%Proposed version
tic;
locD=(1:3)*(D==E.');
for i=1:m
Ai=A(i,:);
Ei=E(i);
Fi=F(i,:);
[~,locB]=ismember(Ai,B);
[~,idx]=max( C(locB,:,locD(i))>Fi(:) ,[],2);
A(i+1,:)= B(idx);
end
toc
Elapsed time is 0.050406 seconds.
A2=A;
isequal(A1,A2)
ans = logical
1
  댓글 수: 1
icdi
icdi 2021년 6월 8일
Thanks! Your code works very well and saves me time. Thanks again.

댓글을 달려면 로그인하십시오.

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Logical에 대해 자세히 알아보기

태그

Community Treasure Hunt

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

Start Hunting!

Translated by