根据行和列索引向量分配新矩阵值

Question

这里是MatLab新手（R2015a，MacOS10.10.5），希望能找到解决这个索引问题的方法。

我想基于一个行索引向量和一个列索引向量来更改一个大型2D矩阵的值。举一个非常简单的例子，如果我有一个3 x 2的零矩阵：

A = zeros(3, 2)

0 0
0 0
0 0

我要把A（1,1）=1，而A（2,2）=1，而A（3,1）=1，这样A现在

1 0
0 1
1 0

我想用向量来表示行和列的索引：

rows = [1 2 3];
cols = [1 2 1];

有没有办法做到这一点而不循环？请记住，这是一个玩具示例，需要处理非常大的2D矩阵。为了获得额外的积分，我是否还可以包含一个向量来指示要插入的值，而不是将其固定为1？

我的循环方法很简单，但很慢：

for i = 1:length(rows)

    A(rows(i), cols(i)) = 1;

end

Answer 1

你基本上得到的是矩阵的稀疏定义。因此，sub2ind的替代方法是sparse。它将创建一个稀疏矩阵，使用full将其转换为完整矩阵。

A=full(sparse(rows,cols,1,3,2))

Answer 2

在检查sub2ind的速度时，我在网上找到了这个答案。

idx = rows + (cols - 1) * size(A, 1);

因此呢

A(idx) = 1 % or b

在一个大矩阵上的5次测试（约5秒的操作）表明它比sub2ind快20%。

这里也有一个n维问题的代码。

Answer 3

sub2ind可以在这里提供帮助，

A = zeros(3,2)
rows = [1 2 3];
cols = [1 2 1];

A(sub2ind(size(A),rows,cols))=1
A =

     1     0
     0     1
     1     0

将向量设置为“插入”

b = [1,2,3];
A(sub2ind(size(A),rows,cols))=b

A =

     1     0
     0     2
     3     0