通过在 MATLAB 中使用 if 语句从组合两个其他矩阵创建新矩阵

Question

我有一个尺寸为 8x6 的矩阵。这个矩阵中有一半的元素是 0，这完全没问题。现在，我想参考另一个 160x6 的矩阵。 8x6 矩阵基于 160x6 矩阵，并来自滚动窗口（20 个观察值）。

我想创建一个新矩阵（同样是 160x6）。每当我的 8x6 矩阵中的一个元素等于 0 时，我希望原始 160x6 矩阵中引用该元素（为 0）的 20 个观察值也为 0。

我尝试了以下方法：

for t=1:T
    for i=1:N
    if B(:,i) == 0;
    C(t,i) = 0;
    else
    C(t,i) = A(t,i);
    end
    end
end

我在哪里：

A 是 160x6 矩阵

B 是 8x6 矩阵

C 是 160x6 矩阵的新输出

目前，我获得了一个“新”的 160x6 矩阵 (C)，但它完全复制了原始的 160x6 矩阵 (A)。所以循环或者if语句是不正确的。

Answer 1

我将根据我对您的问题的理解举一个小例子。

>> B = randi(10,8,6) - 5; % Sample B matrix

B =

    -4     0    -4     4     5     1
    -2     3     3     1     3    -4
    -1     3     1    -3     1     4
     2     5     0    -2     0     4
    -3     4     5     4    -4     3
     3    -1     2    -4     2    -3
    -3     2     2     0    -4     2
     2    -3     4    -3    -4     1

在此矩阵中，您要识别具有0 的位置，例如(1,2), (4,3), (4,5)。并且在 160 x 6 矩阵中 (1:20,2), (61:80,3), (61:80,5) 应该为零。您可以使用repelem 函数来获取此类索引。

>> zeroIdx = repelem(B == 0,20,1)

zeroIdx 将在 B 为零时包含 true，每行重复 20 次。

>> C = A
>> C(zeroIdx) = 0 % Assign 0 to C using zeroIdx

Answer 2

检查以下内容：

%Initialize A matrix with ones for testing.
A = ones(160, 6);
B = ones(8, 6);

%Put few zeros in B
B(1:2:end,1) = 0;
B(5:3:end,3) = 0;

T = 160;
N = 6;

for t=1:T
    for i=1:N
        %The formula k = floor((t-1)/20)+1 equals 1, 1, 1, 1... 20 times, then 2, 2, 2, 2... 20 times
        k = floor((t-1)/20)+1;
        if B(k,i) == 0;
            C(t,i) = 0;
        else
            C(t,i) = A(t,i);
        end
    end
end

%Display C as an image (for testing).
figure;imagesc(C);colormap gray

测试结果图片：

---

k 的值如下图所示：

T=160;t = 1:T;k = floor((t-1)/20)+1;figure;plot(t, k, 'x');grid on;

---

我能实现的最紧凑的解决方案：

C = A.*imresize((B ~= 0), size(A), 'nearest');