Wie wählt man ein Element aus jeder Spalte einer Matrix aus einem neuen Vektor?

Question

Ich habe eine M × N Matrix A, und eine M × 1 Indexvektor ind. Ich möchte einen N × 1 Vektor c erhalten, wobei c(i) = A(ind(i),i) für i = 1, 2, ..., N.

Zum Beispiel lassen

A = hilb(5); 
ind = [2,3,1,4,2]'; 

Wie kann ich Vektor c bekommen?

Answer 1

Das ist, was sub2ind tut:

c = A(sub2ind(size(A), ind(:).', 1:numel(ind))); 

Sie können auch manuell tun, es zu erhöhen Geschwindigkeit ein wenig:

c = A((0:numel(ind)-1) * size(A,1) + ind(:).'); 

Um zu verstehen, wie dies funktioniert, auf linear indexing lesen.