Use values of array as indices for different array

3 visualizzazioni (ultimi 30 giorni)
I have two matrices, A, B, and C, shown below:
A = [16 2 3 13
5 11 10 8
9 7 6 12
4 14 15 1];
B = [1 3
2 3
1 4];
C = [1 3 4];
I want to use the elements of B and C as indices of A to create a new matrix, D, such that each element in C corresponds to a row index in A, and the values in each row of B correspond to column indices given the row index from C. e.g. C(1) = 1 and B(1,:) = [1 3] gives me D(1,:) = A(1,[1 3]). The resulting matrix D would look as follows:
D = [16 3
7 6
4 1];
This is relatively straightforward with a for-loop, but I was hoping for a one-line solution along the lines of A(C,B) = D, which doesn't do what I'm describing in MATLAB. Does anything like what I'm describing exist?

Risposta accettata

madhan ravi
madhan ravi il 16 Mag 2019
D = A(sub2ind(size(A),repmat(C(:),1,size(B,2)),B))

Più risposte (1)

Thomas Augenstein
Thomas Augenstein il 16 Mag 2019
Great, thank you both!

Community Treasure Hunt

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

Start Hunting!

Translated by