Question

我有一个只包含0的6X4矩阵M1。我还有两个1D数组Y1和Y2，每个数组长度为4.这两个数组包含所需的索引值。现在，我想设置（转换为1）矩阵M1的元素，这样 M1(Y1:Y2)等于1

代表：Y1=[1 2 2 1]和Y2=[3 4 5 3]
那么，M1应该是

我可以使用for循环来做到这一点。但有没有优化的方法呢？（我打算使用更大的矩阵）

Answer 1

使用cumsum！

>> szM = size(M1);
>> M1( sub2ind( szM, Y1, 1:szM(2) ) ) = 1
M1 =
 1     0     0     1
 0     1     1     0
 0     0     0     0
 0     0     0     0
 0     0     0     0
 0     0     0     0
>> M1( sub2ind( szM, Y2+1, 1:szM(2) ) ) = -1
M1 =
 1     0     0     1
 0     1     1     0
 0     0     0     0
-1     0     0    -1
 0    -1     0     0
 0     0    -1     0
>> M = cumsum(M,1)
M =
 1     0     0     1
 1     1     1     1
 1     1     1     1
 0     1     1     0
 0     0     1     0
 0     0     0     0

陷阱： 如果Y2中的任何一个等于6而不是将Y2+1设置为-1将超出矩阵维度要解决此问题，您可以在设置为-1的元素之前添加两行：M：

>> cols = 1:szM(2);
>> sel = Y2 < szM(1);
>> M1( sub2ind( szM, Y2(sel)+1, cols(sel) ) ) = -1

使用bsxfun对Pavan Yalamanchili's answer进行衍生:(悬停以查看:)

使用不bsxfun的{{1}}：
offsets

Answer 2

可能还有其他技术，但这会使用疯狂并行的元素操作。

一个非常简单的解决方案。谢谢@ Shai

>> [rows, cols] = size(M);
>> Y1=[1 2 2 1]; Y2=[3 4 5 3]; 
>> M = bsxfun(@ge, (1:rows)', Y1) & bsxfun(@le, (1:rows)', Y2)
M =
     1     0     0     1
     1     1     1     1
     1     1     1     1
     0     1     1     0
     0     0     1     0
     0     0     0     0

不必要的复杂代码

[rows, cols] = size(M);
offsets = ((1 : cols) - 1) * rows
Y1 = offsets + Y1;
Y2 = offsets + Y2;

M = reshape(1:numel(M), rows, cols);
M = bsxfun(@ge, M, Y1) & bsxfun(@le, M, Y2);

在matlab中索引2D数组

2 个答案: