MATLAB: Use multidimensional Indices returned by max to get values from another array of same size

array indexingmultidimensional arrays

I have a three dimensional array and I find the maximum value and indices of these values along the third dimension. How do I use these indices to find the values at the corresponding locations in another array of the same dimensions,
For example:
x = randn(5,10,20)
[xmax,ixmax]=max(x,[],3);
% ixmax is a 5 x 10 array that has the indices of the maxima along the third dimension of x
y = randn(5,10,20);
% I want to find the values of y at the locations of the maxima of x
% I can of course do it by looping
for ii = 1:5
    for jj = 1:10
        yxmax(ii,jj)=y(ii,jj,ixmax(ii,jj));  
    end
end
How do I do this without looping?

Best Answer

If you're using release R2019a or later, specify the 'linear' option to obtain linear indices from max and use those linear indices directly to index into the other array of the same size.
rng default
x = reshape(randperm(27), [3 3 3]);
y = reshape(randperm(27), [3 3 3]);
[value, index] = max(x, [], 'linear');
Let's look at x side-by-side with y, separated by a strip of NaN values.
sideBySide = cat(2, x, NaN(3, 1, 3), y)
Look at the indices, the elements of x, and the corresponding elements of y.
I = index(:);
results = [I, x(I), y(I)]
As an example the second element in I is 4. If you look at sideBySize, the fourth element of x is 16 and the fourth element of y is 8. This is exactly what the second row of results shows.
Related Question