Replacing for loops with vectorization
显示 更早的评论
Questions similar to this have been asked before, but I've struggled to understand the answers. I am looking to speed up my Matlab programs by avoiding for loops. I have seen the remark that Matlab is a high level programming language and so things like for loops shouldn't be necessary, but if anyone could help me figure out how to avoid them in my simple example I would very much appreciate it.
The following script simulates the motion of a pendulum quite well:
length=0.1;
g=9.81;
omega(1)=0;
theta(1)=2;
t(1)=0;
timeStep = 0.01;
t = t(1):timeStep:t(1)+(timeStep*9999);
for i=2:10000
omega(i) = omega(i-1) + (g/length)*sin(theta(i-1))*timeStep;
theta(i) = theta(i-1)+omega(i)*timeStep;
end
plot(t,theta);
xlabel('Time (s)');
ylabel('Theta (rad)');
I have figured out how to vectorize t, shown before the start of the for loop. The issue is that omega and theta are both part of two simultaneous equations, where the next value of theta needs to be calculated based on the previous value of omega, and the next value of omega needs to be calculated based on the previous value of theta.
Is it possible to do this to use vectorization instead of a for loop?
Thank you,
Kyle
P.S. This is for teaching, my students will appreciate anything that speeds up their work!
5 个评论
KALYAN ACHARJYA
2021-2-19
编辑：KALYAN ACHARJYA
2021-2-19
"P.S. This is for teaching, my students will appreciate anything that speeds up their work!"
Slightly higher speed by considering preallocation of the arrays. Suggested avoid the "length" as a variable name, as MATLAB have it's own inbuilt function named as "length"
len=0.1;
g=9.81;
timeStep = 0.01;
t =0:timeStep:timeStep*9999;
omega=zeros(1,length(t));
theta=[2,zeros(1,length(t)-1)];
for i=2:length(t)
omega(i) = omega(i-1) + (g/len)*sin(theta(i-1))*timeStep;
theta(i) = theta(i-1)+omega(i)*timeStep;
end
plot(t,theta);
xlabel('Time (s)');
ylabel('Theta (rad)');
Time in my system
Elapsed time is 0.144906 seconds.
Kyle Baldwin
2021-2-19
"I am looking to speed up my Matlab programs by avoiding for loops. I have seen the remark that Matlab is a high level programming language and so things like for loops shouldn't be necessary"
Sometimes loops are necessary.
Sometimes loops are faster.
Sometimes vectorized code is faster.
It depends on many factors which cannot be summarized into one little nugget of advice like that.
KALYAN ACHARJYA
2021-2-19
编辑：KALYAN ACHARJYA
2021-2-19
@Stephen Cobeldick Finally I have confirmed in this case. Sir, thanks for your valuable advices.
Kyle Baldwin
2021-2-19
采纳的回答
James Tursa
2021-2-20
编辑：James Tursa
2021-2-20
1 个投票
No, in general you cannot vectorize loops such as this. What you are doing in this particular loop is solving a 2nd order differential equation numerically. At each step, the derivative depends on the current state. Thus, you have to iterate through the time steps to integrate from state to state. Since you don't know all of these states ahead of time, you can't vectorize the results with some type of matrix calculation. You will need to have the loop as you are currently doing. Even if you were to call a function such as ode45( ) for this, there would still be iteration loops embedded in ode45( ) to calculate the result.
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Programming 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!
Translated by 
