matrix indexing without loop

2 visualizzazioni (ultimi 30 giorni)
mcdull hall
mcdull hall il 2 Ago 2018
Commentato: mcdull hall il 2 Ago 2018
Hello,
Suppose I have two matrices, A and B:
A = [2 3 1 4;
1 4 1 3];
B = [1 2 3 4;
4 3 2 1];
Now I get the max indices for the matrix A:
[~,ind] = max(A,[],2);
I want to obtain the corresponding values of matrix B with ind:
for ii = 1:2
v(ii) = B(ii,ind(ii));
end
How to get the values of B without a for loop?

Risposta accettata

Walter Roberson
Walter Roberson il 2 Ago 2018
v = B(sub2ind(size(B), 1:size(B,1), ind))
  1 Commento
mcdull hall
mcdull hall il 2 Ago 2018
By the way, it should be:
v = B(sub2ind(size(B), (1:size(B,1))', ind))
Thanks.

Accedi per commentare.

Più risposte (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by