1

accumarray在 Matlab 中非常棒，我经常使用它。我有一个问题，我想传递给的函数accumarray是加权平均值。即它接受两个向量，而不是一个向量。这似乎是不支持的用例accumarray。

我的理解正确吗？

考虑，函数weightedAverage

function [ result ] = weightedMean( values, weights)

result = sum(values(:).*weights(:)) / sum(weights(:));

end

现在，我们要运行accumarray如下：

subs    = [1 1 1 2 2 3 3 3];
vals    = [1 2 3 4 5 6 6 7];
weights = [3 2 1 9 1 9 9 9];

accumarray(subs, [vals;weights],[], @weightedMean)

但matlab返回：

Error using accumarray
Second input VAL must be a vector with one element for each row in SUBS, or a scalar.
4

1 回答 1

1

你是对的，第二个输入必须是列向量或标量。您可以传递一个索引数组，而不是将您的数据传递给accumarray，然后您可以使用这些索引从计算加权平均值的匿名函数中索引到您的values和weights向量。

inds = [1 1 2 2 3 3].';
values =  [1 2 3 4 5 6];
weights = [1 2 1 2 1 2];

output = accumarray(inds(:), (1:numel(inds)).', [], ...
                    @(x)sum(values(x) .* weights(x) ./ sum(weights(x))))
%    1.6667
%    3.6667
%    5.6667
于 2017-01-13T22:13:20.837 回答