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

2 次查看(过去 30 天)
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?

采纳的回答

Steven Lord
Steven Lord 2020-1-10
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.
  1 个评论
JoeB
JoeB 2020-1-10
This solution is preferable because it does not need a call to ndgrid. By ising the linear indexing option then one can apply the index similarly to a 1-D array. Glad to hear about this new option.
In my example this would be as follows:
x = randn(5,10,20);
y = randn(5,10,20);
[xmax,linindex] = max(x,[],3,'linear');
yxmax = y(linindex);
% test by applying index to x
xmaxtest = x(linindex);
plot(xmax,xmaxtest,'.');

请先登录,再进行评论。

更多回答(1 个)

Walter Roberson
Walter Roberson 2020-1-10
When you max() a 3d array along the third dimension, the 2d second output, Index, indexed at J, K is such that Array(J, K, Index(J, K)) is the location of a maximum.
You can construct
[Jgrid, Kgrid] = ndgrid(1:size(Array, 1),1:size(Array, 2));
linidx = sub2ind(size(Array), Jgrid, Kgrid, Index)) ;
And then OtherArry(linidx)

类别

Help CenterFile Exchange 中查找有关 Matrix Indexing 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by