How to replace two nested for with vectorization

2016 3 29
1 个回答
回答已采纳
更新时间：2016 3 29
3 次查看（30 天）

0 个投票

How can I write this without loops
[nrows, ncols] = size(A);
for i = 1:nrows
for j = 1:ncols
if A(i,j)~=0
AC(i) = AC(i)+C(i,A(i,j));
else
break
end
end
end

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

 采纳的回答

Walter Roberson
Walter Roberson 2016-3-29

0 个投票

maxA = max(A(:));
Ccopy = C;
Ccopy(:,maxA+1) = 0;
A(A==0) = maxA+1;
for i = 1 : nrows
AC(i) = sum( C(sub2ind(size(C), 1:nrows, A(:,j))) );
end
Provided that in code before what was posted, AC was initialized to 0's. If AC was not initialized then replace the code with
error('This is the vectorized equivalent to broken code')

更多回答（0 个）

类别

在 帮助中心 和 File Exchange 中查找有关 Language Fundamentals 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by