Vectorization of a for loop to increase speed

1 次查看（过去 30 天）
Howard Wilton
Howard Wilton 2022-11-25
编辑： Bruno Luong 2022-11-25
I am looking to embed an logical expression in an equation. The code example below is to show the problem and is not a reflection of the underlying math I am trying to solve!
t_ = -1:0.1:1;
for t = t_
flag = (t > a); % a is a vector, defined seperately
res = t - a(flag(:));
end
For each value of t, I would get a vector that has a length that corresponds to the number of logical 1s in .
What I would very much like to do is eliminate the for loop and do something kind of like this,
t = -1:0.1:1;
res = t - a(t > a)(:) % i presume this looks AWFUL
And as a result i end up with a matrix with one dimension the size of length t and the other dimension the size of the number of logical 1s in .
Would welcome any ideas or comments.
  1 个评论
Stephen23
Stephen23 2022-11-25
编辑：Stephen23 2022-11-25
"And as a result i end up with a matrix with one dimension the size of length t and the other dimension the size of the number of logical 1s in flag."
Your concept does not have a general solution because the "number of logical 1s in flag" could be different on each loop iteration, so that res vectors have different lengths, and therefore there is no such matrix (because each "other dimension" could be different).
Would you allow for padding of a matrix, or a 1xN vector, or use of a container array?
"The code example below is to show the problem and is not a reflection of the underlying math I am trying to solve!"
It would probably be much faster to explain the actual problem.

请先登录，再进行评论。

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

采纳的回答

Bruno Luong
Bruno Luong 2022-11-25
编辑：Bruno Luong 2022-11-25
This returns a matrix, whose elements which do not satisfy the condition and was not computed in your orginal code are marked by NaN. Each row corresponds to one iteration of the original code
resmat = t_(:) - a(:).';
resmat(resmat<=0) = NaN;
% Move NaN towards the last columns
[~,js] = sort(isnan(resmat),2);
m = size(resmat,1);
resmat = resmat((js-1)*m+(1:m)')

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Mathematics 的更多信息

标签

产品


版本

R2020b

Community Treasure Hunt

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

Start Hunting!

Translated by