matlab - 在 Matlab 中矢量化条件求和循环

Question

我有两个 M×N 矩阵，标记为 A 和 B，我想为 A 的每个唯一值创建一个向量，其中包含 B 中所有项的总和。例如，我有以下矩阵:

A = [6 2 3 4
     5 2 3 3
     5 5 6 2];

B = [.2 .5 .4 .1
     .7 .2 .5 .1
     .6 .6 .1 .9];

我想创建一个向量 C，其中每个索引对应于 D = unique(A) 的索引。所以对于这种情况，D = [2,3,4,5,6] 和

for I = 1:length(D)
    C(I) = sum(B(A(:)==D(I));
end

当 D 的长度为 2000 项并且 A 和 B 都是 ~4000x20 矩阵时，这会变得非常慢。对加快速度有帮助吗？我尝试执行以下操作:

Indxs = bsxfun(@eq,A,reshape(D,1,1,length(D)));
for I = 1:size(Indxs,3)
    C(I) = sum(B(Indxs(:,:,I));
end

但实际上并没有更快。

Answer 1

您可以使用 unique 的常用组合来完成此操作和 accumarray :

[D, ~, uA] = unique(A(:));
C = accumarray(uA, B(:));

使用您的示例数据，结果是:

>> D
D =
     2
     3
     4
     5
     6

>> C
C =
    1.6000
    1.0000
    0.1000
    1.9000
    0.3000

accumarray 的最基本形式(两个输入参数)的作用是对具有相同第一个参数值的所有第二个参数值求和。

---

如果 A 仅包含正整数，则有一个单行解决方案 sparse和 find :

[D, ~, C] = find(sparse(A(:), 1, B(:)));

这是有效的，因为 sparse 会累积对应于相同索引的值。