How to Vectorize if statement

2013 12 2
1 个回答
回答已采纳
更新时间：2013 12 2
11 次查看（30 天）

0 个投票

Dear all,
Here is part of my code for vectorizing the for loop and if statement:
u=zeros(N,1);
du=zeros(n,1);
j=1:1:n
du(rem(j,5)==1)=u(j+1)-u(N+(j-rem(j,5));
du(rem(j,5)~=1)=u(j+1)-u(N+(j-rem(j,5)-1);
but i got the following error:
In an assignment A(I) = B, the number of elements in B and I must be the same.
Can anyone point out the error for me? Thank you.

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

 采纳的回答

Walter Roberson
Walter Roberson 2013-12-2

0 个投票

j5 = rem(j,5);
jsel = j5 == 1;
jnsel = ~jsel;
du(jsel) = u(j(jsel) + 1) - u(N + j(jsel) - j5(jsel));
du(jnsel) = u(j(jnsel) + 1) - u(N + j(jnsel) - j5(jnsel) - 1);
Alternately:
j5 = rem(j,5);
jnsel = j5 ~= 1;
du = u(j + 1) - u(N + j - j5 - jnsel);

更多回答（0 个）

类别

在 帮助中心 和 File Exchange 中查找有关 Loops and Conditional Statements 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by