Is it possible to use an array in an if statement?
1 回表示 (過去 30 日間)
古いコメントを表示
Hi, I need to compare each member of an array against a fixed value and then modify that memeber according to whether it is smaller or larger than the fixed value. I can loop through the array to do this, but I'm trying to speed the code up and was wondering whether you could simply operate on the array directly? I've tried it and I get the same operation on each member irrespective of whether it is smaller or larger than the fixed value.
The code I've used is as follows:
if Gr<10^5
Nu_G=0.5.*Gr.^0.25;
else
Nu_G=0.13.*Gr.^0.33;
end
Can anyone advise whether this is possible or whether there may be any other way to achieve the same effect i.e. avoid the loop and/or speed up the code
Thanks,
Rebecca
0 件のコメント
採用された回答
Daniel Shub
2011 年 11 月 8 日
Nu_G = zeros(size(Gr));
Nu_G(Gr<10^5) = 0.5.*Gr(Gr<10^5).^0.25;
Nu_G(Gr>=10^5) = 0.13.*Gr(Gr>=10^5).^0.33;
This is potentially slower than the loop method. In your if/else you only need to test Gr<10^5 once for every element, in the vectorized version you need to test twice.
その他の回答 (1 件)
Jakob Sievers
2011 年 11 月 8 日
Assuming your GR vector is LGR long, then perhaps something like:
Nu_G=zeros(LGR,1);
Nu_g(Gr<10^5)=0.5.*Gr(Gr<10^5).^25;
Nu_g(Gr>=10^5)=0.13.*Gr(Gr>=10^5).^33;
There's probably even faster ways but this would definitely be faster than doing loops.
5 件のコメント
Daniel Shub
2011 年 11 月 8 日
On my machine:
Gr = randn(1e8, 1);
tic
LGR = length(Gr);
Nu_G=zeros(LGR,1);
Nu_g(Gr<10^5)=0.5.*Gr(Gr<10^5).^25;
Nu_g(Gr>=10^5)=0.13.*Gr(Gr>=10^5).^33;
toc
tic
LGR = length(Gr);
Nu_G=zeros(LGR,1);
for iGr = 1:LGR
if Gr(iGr)<10^5
Nu_g(iGr)=0.5.*Gr(iGr).^25;
else
Nu_g(iGr)=0.13.*Gr(iGr).^33;
end
end
toc
Elapsed time is 104.696162 seconds.
Elapsed time is 41.326568 seconds.
For smaller N, the vectorization is faster.
参考
カテゴリ
Help Center および File Exchange で Matrix Indexing についてさらに検索
製品
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!