累积求和 - MATLAB

Question

假设我有2输入矢量x和相同尺寸

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

和输出y这是在x元素的累加和的reset。每当重置的值对应于1，对于元素的累积和复位，从头再来就像下面

y = [1 3 6 4 9 15] 

我怎么会在Matlab中实现这一点？

Answer 1

一种方法 -

%// 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

这里有一个办法：

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

这工作，因为如果第一个输入accumarray进行排序，每个组的第二输入中的顺序被保存（更多有关这个here）。

与 diff和 cumsum