Matlab组合矩阵取决于条件

Question

我有一个testfile.txt，它是一个 4 x 4 矩阵和制表符分隔

1    1    3    4
2    2    3    4
3    1    3    4
4    2    3    4

我想要的输出是这样的：

1. 如果检测到第二列有 1，则在右侧插入一个新列，新列应该包含类似x=[1 1 0 3]的内容

2. 如果检测到第二列有 2，则在右侧插入新列，新列应包含 y=[2 3 4 5] 之类的内容

输出应该是这样的：

1    1    x=[1    1    0    3]    3    4
2    2    y=[2    3    4    5]    3    4
3    1    x=[1    1    0    3]    3    4
4    2    y=[2    3    4    5]    3    4

最终，在 MATLAB 中，这是我想要得到的输出：

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

我试过的是：

test=dlmread('testfile.txt','\t');
m=length(test);
for i=1:m
    if find(test(:,2)==1)>0
        x=[1    1   0   3];
        test=[test(:,1) x test(:,3:4)];
    elseif find(test(:,2)==2)>0
        y=[2    3    4    5];
        test=[test(:,1) y test(:,3:4)];   
    dlmwrite('testfile.txt',test,'delimiter','\t','precision','%.4f');
    end
end

我得到的错误如下：

被连接的矩阵的维度不一致。

错误来自以下语句：

Error in : test=[test(:,1) x test(:,3:4)]

如果有人可以帮助我，我将非常感激，因为我是 MATLAB 的新手。

提前致谢！

Answer 1

这是一个完全矢量化的解决方案。让我们一步一步来。您显然正在正确读取文本数据，所以让我们保持该代码相同。

test = dlmread('testfile.txt','\t');

我要做的是创建一个二维数组，其中第一行对应于您要插入的x，而第二行对应于您要插入的y。换句话说，创建一个名为 insertData 的变量，这样：

insertData = [1 1 0 3; 2 3 4 5];

接下来，您只需使用第二列来确定您想要将来自insertData 的哪行 数据插入到最终矩阵中。然后，您可以使用它来创建最终矩阵，我们将其存储在 testOut 中。换句话说：

testOut = [test(:,1:2) insertData(test(:,2),:) test(:,3:4)]

我得到的输出是：

testOut =

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

---

让我们慢慢看一下上面的代码。存储在test 中的数据的前两列和存储在test 中的数据的最后两列是相同的。您想在中间插入数据。因此，您创建了一个名为testOut 的新矩阵，其中前两列相同，然后在中间是有趣的地方。每次第二列有 1 时，我们访问insertData 的第一行，并将我们的数据放在相应的行中。每次第二列有 2 时，我们访问insertData 的第二行，并将我们的数据放在相应的行中。要完成所有内容，最后两列应该相同。

---

小提示

如果您想了解为什么您的代码无法正常工作，那是因为您没有正确连接行。此外，在您的for 循环中，您使用: 来访问特定列的所有行，而您应该一次访问一行...至少我是这样解释您的@987654337 @ 环形。此更改也必须在您的 if 语句中完成。此外，当您需要声明一个 NEW 变量时，您正在添加到 test 变量。此外，您需要移动dlmwrite 方法，使其在for 循环完成后被称为AFTER，并且您已经完成了新矩阵的创建。所有这些因素的结合最终导致代码中出现错误。

基本上，如果你想使用你的代码，你需要做的是：

test=dlmread('testfile.txt','\t');
m=length(test);
testOut = []; %// Must declare NEW variable
for i=1:m
    if find(test(i,2)==1)>0 %// Change
        x=[1    1   0   3];
        testOut=[testOut; test(i,1) x test(i,3:4)]; %// NEW
    elseif find(test(i,2)==2)>0 %// Change
        y=[2    3    4    5];
        testOut=[testOut; test(i,1) y test(i,3:4)];   %// NEW
    end
end

%// Move this out!
dlmwrite('testfile.txt',testOut,'delimiter','\t','precision','%.4f');

---

看看testOut 在for 循环中是如何连接的。您将采用testOut 的当前状态，使用; 移动到下一行，然后添加新数据。

这段代码现在应该可以工作了，但您只需两行就可以轻松实现您想要做的事情。

---

希望这有帮助！