Indexing multidimensional array with a matrix in Matlab

Question

How can I index the dimensions of a nd-array element-wise by using a 2d matrix which entries represent which dimensions (representing slices or 2d Matrices) to take the values from?

I=ones(2)*2;
J=cat(3,I,I*2,I*3);

indexes = [1 3 ; 2 2] ;

so J is

J(:,:,1) =

 2     2
 2     2


J(:,:,2) =

 4     4
 4     4


J(:,:,3) =

 6     6
 6     6

it works easily using 2 for loops

for i=1:size(indexes,1)
     for j=1:size(indexes,2)
        K(i,j)=J(i,j,indexes(i,j));
    end
end

which yields the desired result

but is there a vectorized / smart indexing way of doing this?

%K=J(:,:,indexes)  --does not work

Jonas · Accepted Answer · 2014-01-29 11:10:05Z

3

Just use linear indexing:

nElementsPerSlice = numel(indexes);

linearIndices = (1:nElementsPerSlice) + (indexes(:)-1) * nElementsPerSlice;

K = J(linearIndices);

answered Jan 29, 2014 at 11:10

Jonas

75.1k10 gold badges139 silver badges179 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

wanjolivet · Accepted Answer · 2015-06-17 17:58:34Z

1

you can use sub2ind to convert matrix indexes into linear index

ind = sub2ind( size(J), [1 1 2 2], [1 2 1 2], [1 3 2 2]);
K = resize(J(ind), [2 2]);

answered Jun 17, 2015 at 17:58

wanjolivet

914 bronze badges

Collectives™ on Stack Overflow

Indexing multidimensional array with a matrix in Matlab

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related