Стабильный аккумулятор в MATLAB

Мы можем использовать sortrows в качестве шага предварительной обработки, чтобы сначала отсортировать индексы и соответствующие значения, как указано в документации:

SORTROWS использует стабильную версию быстрой сортировки.

Поскольку индексы в subs должны быть отсортированы относительно их линейных индексов, нам нужно отсортировать их в обратном лексикографическом порядке. Этого можно добиться, изменив порядок столбцов до и после использования sortrows.

Это дает нам следующий код для стабильной версии accumarray:

function A = accumarrayStable(subs, val, varargin)
[subs(:,end:-1:1), I] = sortrows(subs(:,end:-1:1));
A = accumarray(subs, val(I), varargin{:});

Альтернатива:

Как было предложено Луисом Мендо, вместо sortrows можно также сгенерировать линейные индексы из индексов и использовать вместо них sort.

function A = accumarrayStable(subs, val, varargin)
if numel(varargin)>0 && ~isempty(varargin{1})
    sz = varargin{1};
else
    sz = max(subs,[],1);
end
[~, I] = sort(subs*cumprod([1,sz(1:end-1)]).');
A = accumarray(subs(I,:), val(I), sz, varargin{:});

Обратите внимание, что мы должны использовать 1+(subs-1)*cumprod([1,sz(1:end-1)]).' для преобразования в линейные индексы. Мы опускаем +1 и -1, так как результат sort будет таким же; что экономит нам несколько циклов.

Какое из вышеперечисленных решений будет быстрее, будет зависеть от вашей машины и версии MATLAB. Вы можете протестировать, например, через:

A = randi(10, 1e4, 5); 
timeit(@()accumarrayStable(A(:,1:end-1), A(:,end), [], @(x)x(1))