How to vectorize this nested for loop?

1 次查看（过去 30 天）
Awanish
Awanish 2024-3-2
回答： Dyuman Joshi 2024-3-2
Hi everyone, I want to vectorize this nested for loop with an 'if' condition. Could you please let me know how to get it done?
b=0;
pair = [];
min1 = 1;
max1 = 1;
for n = -min1:max1
for m = -min1:max1
if n+m == b
pair = [pair;n,m];
end
end
end

请先登录，再回答此问题。

采纳的回答

Dyuman Joshi
Dyuman Joshi 2024-3-2
Ran in:
%Original Approach
b=0;
min1 = 1;
max1 = 1;
pair = [];
for n = -min1:max1
for m = -min1:max1
if n+m == b
pair = [pair;n,m];
end
end
end
pair
pair = 3×2
-1 1 0 0 1 -1
%Vectorized approach
[n,m] = meshgrid(-min1:max1)
n = 3×3
-1 0 1 -1 0 1 -1 0 1
m = 3×3
-1 -1 -1 0 0 0 1 1 1
idx = n+m==b
idx = 3×3 logical array
0 0 1 0 1 0 1 0 0
out = [n(idx) m(idx)]
out = 3×2
-1 1 0 0 1 -1

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Loops and Conditional Statements 的更多信息

标签

产品


版本

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by