是否有一个accumarray()将矩阵作为“val”？

Question

accumarray() 的 val 参数必须是一个向量。就我而言，我需要对矩阵的列进行求和（或平均）。有没有一个函数或方法可以实现这一点？

我现在正在做的是在 for 循环中，我分别对列值求和：

for iCol = 1:nCols
    means(:,iCol) = accumarray(labels', X(:,iCol));
end

Answer 1

一种解决方案是复制标签中的行索引并添加另一列列索引。然后，您可以将 X 重塑为列向量并应用 accumarray 一次：

labels = [repmat(labels(:),nCols,1) ...            % Replicate the row indices
          kron(1:nCols,ones(1,numel(labels))).'];  % Create column indices
totals = accumarray(labels,X(:));  % I used "totals" instead of "means"

它是如何工作的...

A = Accumarray(subs,val) 对于列向量 subs向量 val 的工作原理是将 val(i) 中的数字添加到输出列向量 的行 subs(i) 中的总数>A。但是，subs 不仅仅可以包含行索引。它可以包含多个维度的下标索引，以便在输出中为其赋值。此功能允许您处理作为矩阵而不是向量的输入val。

首先，可以使用 冒号运算符 X(:)。接下来，为了跟踪输出中的哪一列应放置 X(:) 中的值，我们可以修改输入 subs 以包含额外的列索引。为了说明其工作原理，我将使用以下示例输入：

labels = [3; 1; 1];
X = [1 2 3; ...
     4 5 6; ...
     7 8 9];
nCols = 3

以下是上述代码中的变量最终的样子：

labels = 3 1    X(:) = 1    totals = 11 13 15
         1 1           4              0  0  0
         1 1           7              1  2  3
         3 2           2
         1 2           5
         1 2           8
         3 3           3
         1 3           6
         1 3           9

例如，请注意，最初的值 1 4 7 X 第一列中的值只会累积在输出的第一列中，如 labels 第二列前三行中的值所示。生成的输出应该与使用问题中的代码（循环遍历每列以执行累积）得到的输出相同。

Answer 2

也许是从 MATLAB 答案（原始答案假设列主输入，因此我转置它们）：

[xx, yy] = ndgrid(labels,1:size(X, 1));
totals = accumarray([yy(:) xx(:) ], reshape(X.', 1, []));

示例：

X = [1 2 3 4; 5 6 7 8];
labels = [2; 1; 3; 1];

给出
总计 = [6 1 3; 14 5 7]。

如果您想逐行执行此操作，则无需转置，只需：

[xx, yy] = ndgrid(labels,1:size(X, 2));
totals = accumarray([xx(:) yy(:)], X(:));