如何矢量化這個for循環

Question

如何矢量化下面的代碼？

這裏x_cords和y_cords都是7894 * 1向量和buffImg是一個虛擬零圖像，我試圖映射線和得到一個細分圖像。

for i = 1:length(x_cords) 
    buffImg(y_cords(i),x_cords(i)) = 1; 
end 

Answer 1

在這種情況下，你需要變換存儲在x_cords和y_cords到使用sub2ind -function線性指標的下標索引，然後您可以直接分配像這樣的：

buffImg=zeros(100,100); 
x=randperm(100); 
y=randperm(100); 
buffImg(sub2ind(size(buffImg),x,y))=1; 

只是爲了告訴你，輸出是一樣的，這裏是你如何測試它：

x=randperm(100); 
y=randperm(100); 
buffImg=zeros(100,100); 
buffImg2=zeros(100,100); 
for i = 1:length(x) 
    buffImg(x(i),y(i)) = 1; 
end 
buffImg2(sub2ind(size(buffImg),x,y))=1; 
all(all(buffImg==buffImg2))