Matlab - 当先前数据的累积和大于特定数字时将数字转换为零

Question

我有一个包含两列的矩阵。

第一列是一组由 0 和 1 组成的数据。

column tow 是一组每天的时间数据。

下面是一个示例数据：

0 2020-01-01
1 2020-01-01
1 2020-01-01
0 2020-01-01
1 2020-01-01
1 2020-01-01
0 2020-01-01
1 2020-01-02
0 2020-01-02
...

当一天内先前数据的累积量大于特定数字时，我想将以下数据设置为零。

根据示例数据，如果具体数字设置为2，则将2020-01-01的数据转换为如下数据：

0 2020-01-01
1 2020-01-01
1 2020-01-01
0 2020-01-01
0 2020-01-01
0 2020-01-01
0 2020-01-01

这是我如何在一天内完成的：

for i = 1:7
    if cumsum(data(1:i,1) > 2)
      then data(i) = 0
    end
end

但我还没有弄清楚如何递归地做到这一点。

递归的意思是当总和大于特定数字时，第二天的数据会重新计算总和，并将后面的数据转换为0。

根据上面的例子：

    1 2020-01-01
    1 2020-01-01
    0 2020-01-01
    1 2020-01-02
    0 2020-01-02
    0 2020-01-02
    1 2020-01-02
    1 2020-01-02
    0 2020-01-02
    1 2020-01-02
    ...

第二天的数据会这样转换：

    0 2020-01-01
    1 2020-01-01
    1 2020-01-01
    0 2020-01-01
    0 2020-01-01
    0 2020-01-01
    0 2020-01-01
    1 2020-01-02
    0 2020-01-02
    0 2020-01-02
    1 2020-01-02
    0 2020-01-02
    0 2020-01-02
    0 2020-01-02
    ...

你可以看到两个不同日期的数据是用同一个进程分开处理的。

我只知道必须将cumsum设置为零并在日期更改时重新计算cumsum。

有没有办法递归呢？

非常感谢！

油箱

Answer 1

我认为在你的情况下不需要递归，简单的矩阵索引会为你做：

my_mat =

 0     8
 1     8
 1     8
 0     8
 1     8
 1     8
 0     8
 1     8
 0     8

让我们以my_mat 作为矩阵，并使用索引对其进行操作：

my_mat((cumsum(my_mat(:,1))>2),1) = 0

输出：

my_mat =

 0     8
 1     8
 1     8
 0     8
 0     8
 0     8
 0     8
 0     8
 0     8

让我们分开命令：

cumsum(my_mat(:,1))

产生一个二进制向量，当条件满足时为真，否则为 0，

0
0
0
0
1
1
1
1
1

现在我们可以通过使用这个向量作为向量索引来访问矩阵中的相关条目。

现在，如果您想进行下一步并仅在第二列为 8 时将值更改为 0，我们将像这样改进索引向量中的条件：

my_mat((cumsum(my_mat(:,1))>2 & my_mat(:,2)==8))=0