# Cumulative sum with conditions

38 views (last 30 days)

Show older comments

I am counting a number of 1's in an array.

But I want to count the cumulative sums only when the number of 1's are followed by a sequence of zeros.

% For instance,

h1=[1 0 0]

h2=[1 1 0]

h3=[1 1 1]

h4=[0 0 1]

h5=[0 1 0].

I want to count the cumulative sums (S) of those arrays as follows.

% The ideal outputs are:

S1= 1

S2= 2

S3= 0

S4= 0

S5= 0

since h2: h4 do not meet the condition of 1's followed by 0's.

I would truly appreciate if any one could advise me an effective way of programming the conditional sums.

##### 1 Comment

Rik
on 3 Oct 2018

### Accepted Answer

Bruno Luong
on 4 Oct 2018

Edited: Bruno Luong
on 4 Oct 2018

Change the last line to make it can handle with single row:

m = size(h,1);

d = diff(h,1,2);

[r,c] = find(d==-1);

b = sum(abs(d),2)==1;

s = accumarray(r(:),c(:),[m 1]).*double(b)

This because find() rotates the result for row input array.

### More Answers (4)

Sean de Wolski
on 3 Oct 2018

h = [1 0 0;

1 1 0;

0 0 1;

1 1 1;

0 1 0];

s = sum(h, 2).*~h(:,end)

requiring the last element to be zero. This wouldn't work for [1 0 1] but I don't know what you'd want for that.

##### 0 Comments

Bruno Luong
on 4 Oct 2018

Edited: Bruno Luong
on 4 Oct 2018

h = [1 0 0;

1 1 0;

0 0 1;

1 1 1;

0 1 0];

[a,c] = max(1-h,[],2);

s = (c-1).*a.*h(:,1)

Bruno Luong
on 4 Oct 2018

Edited: Bruno Luong
on 4 Oct 2018

m = size(h,1);

d = diff(h,1,2);

[r,c] = find(d==-1);

b = sum(abs(d),2)==1;

s = accumarray(r(:),c(:),[m 1]).*double(b)

### See Also

### Categories

### Community Treasure Hunt

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

Start Hunting!