Vectorizing a loop for cumulative sum with reset

4 次查看(过去 30 天)
The lines below work fine to compute cumulative sum on a 1D logical array ('s') such that the summation resets to zero whereever the element in 's' is zero. Question is, how can this be vectorised to avoid loop.
r(1) = s(1); %s is the source vector, r is the result
for i = 2:length(s)
r(i) = (r(i-1) + s(i))*s(i);
end
Thanks.

回答(1 个)

Lars Nelson
Lars Nelson 2023-4-26
Unfortunately, because each value in vector r depends on the previous value in r, I do not think that there is a way to vectorize this code and you will have to use a loop.

类别

Help CenterFile Exchange 中查找有关 Loops and Conditional Statements 的更多信息

产品


版本

R2023a

Community Treasure Hunt

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

Start Hunting!

Translated by