matlab - 向量化下标到矩阵的转换

Question

给定两个向量 r 和 c ，它们将行和列下标保存到矩阵 A 中（也给定了它的大小），我想计算 A。点可以出现多次，并且每次出现都应该增加 A 中的相应元素。一个例子：

r = [1 2 4 3 2];
c = [2 2 1 1 2];
A = zeros(4, 3);

ind = sub2ind(size(A), r, c);
for i = 1 : length(ind)
    A(ind(i)) = A(ind(i)) + 1; % could as well use r and c instead of ind ...
end

这产生了矩阵

A =
     0     1     0
     0     2     0
     1     0     0
     1     0     0

如果可能的话，我想避免循环。这个问题有矢量化的解决方案吗？最好是一种不会产生巨大临时矩阵的...

Answer 1

这是一份工作accumarray：

r = [1 2 4 3 2];
c = [2 2 1 1 2];

%# for every r,c pair, sum up the occurences
%# the [4 3] is the array size you want
%# the zero is the value you want where there is no data
accumarray([r',c'],ones(length(r),1),[4 3],@sum,0)

ans =

     0     1     0
     0     2     0
     1     0     0
     1     0     0

请注意，如果您的结果数组有这么多零（即非常稀疏），sparse则可能是更好的选择，正如@woodchips 所建议的那样

sparse(r,c,ones(size(r)),4,3)

ans =

   (3,1)        1
   (4,1)        1
   (1,2)        1
   (2,2)        2