Selecting Elements of a Matrix by Indexing with a Vector

3 vues (au cours des 30 derniers jours)
Darren Wethington
Darren Wethington le 23 Avr 2020
Commenté : Ameer Hamza le 23 Avr 2020
I have a 100x3 matrix and I'd like to select one element from each row to form a 100x1 vector. Given a vector of 100 indeces between 1-3, how can I obtain this? The code below is what I thought should work, but instead produces a 100x100 matrix (concatenation of each 100x1 vector generated by each value of indx).
A = rand(100,3);
indx = randi([1 3],100,1); % for each row, which column do I want to pull the element from?
B = A(:,indx); % B is 100x100, I want 100x1

Réponse acceptée

Ameer Hamza
Ameer Hamza le 23 Avr 2020
Modifié(e) : Ameer Hamza le 23 Avr 2020
See sub2ind() function.
A = rand(100,3);
indx = randi([1 3],100,1);
idx = sub2ind(size(A), (1:size(A,1)).', indx);
B = A(idx);
  2 commentaires
Darren Wethington
Darren Wethington le 23 Avr 2020
This worked, I was just hoping to not have to sub2ind it and was curious if there was a more straightforward way. I'll use this, thanks.
Ameer Hamza
Ameer Hamza le 23 Avr 2020
I think sub2ind is the simplest way to do such a thing. Following avoids sub2ind, but I am not sure you can call it straightforward
A = rand(100,3);
indx = randi([1 3],100,1);
ind = (1:size(A,1)).' + (indx-1).*size(A,1);
B = A(ind);

Connectez-vous pour commenter.

Plus de réponses (1)

David Hill
David Hill le 23 Avr 2020
A = rand(100,3);
indx = randi([1 3],100,1);
B = A(1:100,indx);
  1 commentaire
Darren Wethington
Darren Wethington le 23 Avr 2020
This did not work for me, I thought it should.

Connectez-vous pour commenter.

Catégories

En savoir plus sur Matrix Indexing dans Help Center et File Exchange

Tags

Community Treasure Hunt

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

Start Hunting!

Translated by