matlab - 对索引求和

Question

假设我有

A = [10 20 30 40];
idx = [1 1 1 2];
result = [0 0];

我需要对 idx 中的索引求和，以便

result(1) = A(1) + A(2) + A(3);
result(2) = A(4);

我实现了代码

for i=1:length(idx)
  result(idx(i)) += A(i);
end

如果可能的话，我怎样才能将它转换为更多 Octave 标准代码？

Answer 1

看看accumarray ，它完全符合您的要求，它只需要将其第一个输入作为列:

A = [10 20 30 40];
idx = [1 1 1 2];
result = accumarray(idx',A)

  result =

      60
      40

是的，这也适用于 Octave ;) ( link )