在分组数据中按窗口大小对数据求和

Question

我对 matlab 很陌生，我正在尝试找到一种方法来完成以下任务，而不使用 for 循环：

我有一个如下所示的数据集：

data = [1 5; 1 3; 1 8; 2 1; 2 2; 2 5; 3 3; 3 8; 3 4]

第一列是一个组（是未来的月年组合）

现在我想计算具有给定窗口大小的第二列的总和，但前提是组索引相同 - 如果不是，则应计算该组中的最大总和。

如果窗口大小=2，我想创建以下结果：

summed_data = [1 8; 1 11; 1 8; 2 3; 2 7; 2 5; 3 11; 3 12; 3 4]

窗口大小为 3 时，结果如下所示：

summed_data = [1 16; 1 11; 1 8; 2 8; 2 7; 2 5; 3 15; 3 12; 3 4]

等等。

我曾考虑通过创建足够的子索引来使用 accumarray - 但我遇到了窗口大小问题并且总和重叠。

有没有人知道如何在不使用循环的情况下实现它？

提前致谢并致以最诚挚的问候 斯蒂芬

Answer 1

这似乎有效：

ws = 2; k = [ones(ws,1);zeros(mod(ws,2),1)];
C = accumarray(data(:,1),data(:,2),[],@(v){conv(v,k,'same')})

您似乎锚定到窗口中的当前像素并向前看。对吗？

不确定这是否涵盖所有极端情况，但它可能会引导您朝着正确的方向前进。

测试：ws = 2

ws = 2; k = [ones(ws,1);zeros(mod(ws,2),1)];
C = accumarray(data(:,1),data(:,2),[],@(v){conv(v,k,'same')});
summed_data = [data(:,1) vertcat(C{:})]
summed_data =

     1     8
     1    11
     1     8
     2     3
     2     7
     2     5
     3    11
     3    12
     3     4

测试：ws = 3

summed_data = [data(:,1) vertcat(C{:})]
summed_data =

     1    16
     1    11
     1     8
     2     8
     2     7
     2     5
     3    15
     3    12
     3     4