Adding values in an array within limits

Hi all,

I have a vector A=[1 2 3 4 5 4]'

I would like to get from vector A, a vector B where its elements are the same with A, until the cumulative sum becomes 9. When the limit of 9 is reached the rest of the elements will be zero.

So the vector B will be : B=[1 2 3 3 0 0]'

I don't know if I explained it well

thanks in advance

Nikolas

Answer 1

[Adam]

B = zeros( size( A ) );
validIdx = cumsum( A ) <= 9;

B( validIdx ) = A( validIdx );
B( nnz( validIdx ) + 1 ) = 9 - sum( B );

would work, assuming you just clip the final value so you have a sum of 9.

Answer 2

[Stephen23]

>> A = [1;2;3;3;5;4];
>> B = A.*(cumsum(A)<=9)
B =
   1
   2
   3
   3
   0
   0

[Nikolas Spiliopoulos]

Basically my vector is A = [1;2;3;4;5;4]; not A = [1;2;3;3;5;4];

thanks!

[Stephen23]

@Nikolas Spiliopoulos: you can use whatever vector you want.

I just used the vector that fitted your desired output, which you gave as "B=[1 2 3 3 0 0]'". Did I guess wrong? If you could please explain the reason for why the fourth elements of A and B should be different, then I can show you how to achieve what you want.