Matlab：使用索引增加矩阵值答案

【问题标题】：Matlab: Increment of matrix values with indicesMatlab：使用索引增加矩阵值
【发布时间】：2018-02-10 18:29:43
【问题描述】：

我有一个索引向量，并希望在每个索引中增加矩阵中的值。例如：

    ind = [1 2 2 5];
    m = zeros(3);
    m(ind) = m(ind) + 1;

结果如下：

    m = [1 0 0
         1 1 0
         0 0 0]

但我需要的结果是

    m = [1 0 0
         2 1 0
         0 0 0]

时间复杂度对我来说很重要，我用不上。谢谢。

【问题讨论】：

请注意，有时，矢量化解决方案并不比 for 循环快，因为 MATLAB 在最近的版本中对循环进行了极大的优化
@AnderBiguri 即使在这种情况下？
最好的了解方式是尝试

标签： matlab matrix increment

【解决方案1】：

这是一种方法。我还没计时。

ind = [1 2 2 5];
N = 3;
m = full(reshape(sparse(ind, 1, 1, N^2, 1), N, N));

同样地，你可以使用

ind = [1 2 2 5];
N = 3;
m = reshape(accumarray(ind(:), 1, [N^2 1]), N, N);

或其变体（感谢@beaker）

ind = [1 2 2 5];
N = 3;
m = zeros(N);
m(:) = accumarray(ind(:), 1, [N^2 1]);

这个可能比其他的慢：

ind = [1 2 2 5];
N = 3;
m = zeros(N);
[ii, ~, vv] = find(accumarray(ind(:), 1));
m(ii) = vv;

【讨论】：

神奇的矢量化解决方案！但是，我感觉第一个肯定会比 for 循环花费更多时间

【解决方案2】：

对于有序的索引数组，我们可以玩diff -

out = zeros(M,N);  % Output array of size(M,N)
df = diff([0,ind,ind(end)+1]);
put_idx = diff(find(df)); % gets count of dups
out(ind(df(1:end-1)~=0)) = put_idx;

基本思想是我们使用diff 沿长度计算重复项。这些计数是要分配给 zeros 数组的值。要分配这些值的索引只是唯一索引，可以通过查找每组重复索引的开始来找到。

基准测试

创建排序索引数组的脚本 (create_data.m) -

function ind = create_data(M,N, num_unq_ind, max_repeats)

unq_ind = unique(randi([1,M*N],1,num_unq_ind));
num_repeats = randi(max_repeats, [1,numel(unq_ind)]);
ind = repelem(unq_ind, num_repeats);

基准测试脚本 (bench1.m) 以测试各种场景 -

clear all; close all;

M = 5000; % Array size
N = 5000;

% Input params and setup input indices array (edited for various runs)
num_unq_ind = 100000;
max_repeats = 100;
ind = create_data(M,N, num_unq_ind, max_repeats);

num_iter = 100; % No. of iterations to have reliable benchmarking 
disp('Input params :')
disp(['num_unq_ind = ' int2str(num_unq_ind)])
disp(['max_repeats = ' int2str(max_repeats)])

disp('------------------ Using diff ----------------')
tic
for i=1:num_iter
    out = zeros(M,N);
    df = diff([0,ind,ind(end)+1]);
    put_idx = diff(find(df));
    out(ind(df(1:end-1)~=0)) = put_idx;
end
toc

% Luis's soln
disp('------------------ Using accumaray ----------------')
tic
for i=1:num_iter
    m = reshape(accumarray(ind(:), 1, [N^2 1]), N, N);
end
toc

各种场景运行 -

>> bench1
Input params :
num_unq_ind = 10000
max_repeats = 10
------------------ Using diff ----------------
Elapsed time is 0.948544 seconds.
------------------ Using accumaray ----------------
Elapsed time is 1.502658 seconds.
>> bench1
Input params :
num_unq_ind = 100000
max_repeats = 10
------------------ Using diff ----------------
Elapsed time is 1.784576 seconds.
------------------ Using accumaray ----------------
Elapsed time is 1.533280 seconds.
>> bench1
Input params :
num_unq_ind = 10000
max_repeats = 100
------------------ Using diff ----------------
Elapsed time is 1.315998 seconds.
------------------ Using accumaray ----------------
Elapsed time is 1.391323 seconds.
>> bench1
Input params :
num_unq_ind = 100000
max_repeats = 100
------------------ Using diff ----------------
Elapsed time is 6.180565 seconds.
------------------ Using accumaray ----------------
Elapsed time is 3.576154 seconds.

由于更少稀疏和更多重复，accumarray 似乎做得更好。

【讨论】：

【解决方案3】：

您可以使用histcounts

n = 3;
m = reshape(histcounts(ind, [1:n^2 n^2]), n, n);

【讨论】：