针对循环优化matlab

Question

我对代码（计算直方图）有以下循环。我正在用Matlab编写。由于我是Matlab新手，我不知道任何更快的方法来做到这一点。我正在写

for i=1:size(b) 
    a(b(i)) = a(b(i)) + 1; 
end 

有没有更快的方法来做到这一点，最好是那些不需要for循环？

Answer 1

您可以简单地通过a(b) = a(b) + 1进行矢量化。检查以下内容：

>> a = [1 2 3 4]; 
>> b = [2 4]; %# indices to modify. Be sure that they are in bounds. 
>> a(b) = a(b) + 1 

a = 

    1  3  3  5 

如果你使用一些指数多次，然后accumarray将有助于如下：

>> a = [1 2 3 4]; 
>> b = [2 4 2]; 
>> a = accumarray([1:numel(a) b].',[a ones(size(b))])' 

a = 

    1  4  3  5 

或者，你可以使用：

>> a = [1 2 3 4]; 
>> b = [2 4 2]; 
>> b = accumarray(b.',ones(size(b))); 
>> a(nzIndex) = a(nzIndex) + b(nzIndex)' 

a = 

    1  4  3  5 

见this nice answer这里更多细节。

Answer 2

如果a和b都是矢量，则应该起作用。

a = 1:100; 
b = 100:-1:1; 

a(b) = a(b) + 1;