Vectorize nested for loops
Show older comments
Hi, I would like to vectorize the following loop, have tried generating indices and but so far been successful.
a = any 1D vector
N = some value
for i=1:length(a)
for j=(i+1):length(a)
if(a(i)-a(j) > N)
disp('Far');
end
end
end
Does anyone have any ideas on this ?
2 Comments
Guillaume
on 31 Jan 2018
The ability of vectorising loops depends solely on the do something. So if you don't tell us what it is, we can't answer your question.
Balkrishna Patankar
on 31 Jan 2018
Answers (1)
Guillaume
on 31 Jan 2018
Your example is trivially vectorised:
%R2016b or later:
isgreater = (a - a.') > N;
%any version:
isgreater = bsxfun(@minus, a, a.') > N;
isgreater(r, c) is true (1) when a(r)-a(c) > N
3 Comments
Balkrishna Patankar
on 1 Feb 2018
Indeed, my answer was equivalent to having the j loop starting at 1 instead of i+1. The upper triangle of that isgreater matrix is the exact equivalent of your loops, so:
isgreater = triu((a - a.') > N, 1)
Balkrishna Patankar
on 2 Feb 2018
Categories
Find more on Loops and Conditional Statements in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!