kümülatif toplamı - MATLAB

Question

varsayalım I2 giriş x ve aynı boyutta

x = [1 2 3 4 5 6] 
reset = [0 0 0 1 0 0] 

ve x elemanların birikimli toplam bir çıkış y arasında reset vektörleri vardır. sıfırlamalarının değer 1'e tekabül zaman, elemanlar için toplu toplamı sıfırlamak ve sadece

y = [1 3 6 4 9 15] 

nasıl Matlab bu uygulayacağını aşağıda gibi herşeye yeniden başlamak?

Answer 1

Bir yaklaşım -

%// Setup few arrays: 
cx = cumsum(x)   %// Continuous Cumsumed version 
reset_mask = reset==1 %// We want to create a logical array version of 
         %// reset for use as logical indexing next up 

%// Setup ID array of same size as input array and with differences between 
%// cumsumed values of each group placed at places where reset==1, 0s elsewhere 
%// The groups are the islands of 0s and bordered at 1s in reset array. 
id = zeros(size(reset)) 
diff_values = x(reset_mask) - cx(reset_mask) 
id(reset_mask) = diff([0 diff_values]) 

%// "Under-compensate" the continuous cumsumed version cx with the 
%// "grouped diffed cumsum version" to get the desired output 
y = cx + cumsum(id) 

Answer 2

İşte yolu:

result = accumarray(1+cumsum(reset(:)), x(:), [], @(t) {cumsum(t).'}); 
result = [result{:}]; 

accumarray ilk giriş sıralanır eğer, ikinci girişin her grup içindeki düzen (bu here hakkında daha fazla) korunur için bu çalışır. diff ve cumsum ile