Wie ändert man Elemente einer Matrix mit Bezug auf einen Vektor von Spaltenindizes ohne For-Schleife?

Question

Ich habe a Matrix

a = 

    0 0 0 0 0 
    0 0 0 0 0 
    0 0 0 0 0 
    0 0 0 0 0 
    0 0 0 0 0 
    0 0 0 0 0 

und b Vektor

b = 

    1 2 3 4 5 5 

Ich mag Wert jeder Zeile mit Referenzwert von b Matrixwert in a Matrix ersetzen, und schließlich eine Matrix zu erzeugen, wie folgt, ohne die Verwendung von für die Schleife.

a_new = 

    1 0 0 0 0 
    0 1 0 0 0 
    0 0 1 0 0 
    0 0 0 1 0 
    0 0 0 0 1 
    0 0 0 0 1 

wenn erstes Element b, b(1) = 1 so nimmt erste Reihe von a Vektors ändern und als erstes Element 1, da b(1) = 1 machen.

Wie kann ich dies ohne die Verwendung von for-Schleife implementieren?

Answer 1

Sicher. Sie brauchen nur einen linear index von b zu bauen und es verwenden, um die Werte in a zu füllen:

a = zeros(6,5); % original matrix 
b = [1 2 3 4 5 5]; % row or column vector with column indices into a 
ind = (1:size(a,1)) + (b(:).'-1)*size(a,1); % build linear index 
a(ind) = 1; % fill value at those positions 

Answer 2

jedes dieser Verfahren arbeitet in Octave:

bsxfun(@eq, [1:5 5]',(1:5)) 

[1:5 5].' == (1:5) 

Answer 3

auch über den Index auf Indizes Umwandlung Art und Weise,

a = zeros(6,5); 
b = [1 2 3 4 5 5]; 
idx = sub2ind(size(a), [1:6], b); % 1:6 just to create the row index per b entry 
a(idx) = 1 

Answer 4

Gleiche wie Luis Mendo's answer, aber mit der speziellen Funktion sub2ind:

a(sub2ind(size(a),(1:numel(b)).',b(:))) = 1