NEW MATRIX WITH IF CONDITIONS

Hello everyone

I have the following 2 matrices in which S1 is randomly developed.I need to develop another matrix D2 in which if S1(i)=1 and while S1(i) stays 0 for the next positions, D2(i) should be added until S1(i)=1 again as shown.

Please help me on this

D1=[10 20 30 40]

S1=[1 0 0 1]

D2=[60 0 0 40]

Answer 1

[Akira Agata]

Like this?

D1 = [10 20 30 40];
S1 = [1 0 0 1];
D2 = zeros(size(D1));

pt = find(S1);

for kk = 1:numel(pt)
  if kk < numel(pt)
      D2(pt(kk)) = sum(D1(pt(kk):pt(kk+1)-1));
  else
      D2(pt(kk)) = sum(D1(pt(kk):end));
  end
end

[PRANAY DISHAN]

Thank you sir. It was very helpful.

Answer 2

[Jos (10584)]

No need for loops or ifs:

% data
D1 = [ 10 20 30 40 50 60]
S1 = [  1  0  0  1  1  0]

% engine
D2 = zeros(size(D1))
D2(S1==1) = accumarray(cumsum(S1(:)), D1)
% D2 = [60 0 0 40 110 0]

[PRANAY DISHAN]

Thank u sir, it will shorten my program.