MATLAB tối ưu hóa cho vòng lặp

Question

Tôi có vòng lặp sau đây cho một mã (tính toán biểu đồ). Tôi đang viết ở Matlab. Vì tôi mới làm quen với Matlab, tôi không biết có phương pháp nào nhanh hơn để làm điều này. Tôi hiện đang viết

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

Có phương pháp nào nhanh hơn để làm điều này, tốt nhất là những phương pháp không yêu cầu vòng lặp?

Answer 1

Bạn chỉ có thể vector hóa nó theo a(b) = a(b) + 1. Kiểm tra như sau:

>> 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 

Nếu bạn sử dụng một số chỉ số nhiều lần, sau đó accumarray sẽ giúp như sau:

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

a = 

    1  4  3  5 

Ngoài ra, bạn có thể sử dụng:

>> 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 

Xem this nice answer đây để biết thêm chi tiết.

Answer 2

Nếu cả hai a và b là vectơ, điều này sẽ hoạt động.

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

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