how do i use the indices from [M,I]=max(A,[],3) to find the corresponding values in another 3D array, B, of the same size as A?

Question

Nicolai Sanders 2016-11-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/310346-how-do-i-use-the-indices-from-m-i-max-a-3-to-find-the-corresponding-values-in-another-3d-array

评论： Nicolai Sanders 2016-11-3

I have two 3d arrays, A and B, of size n,m,k.

I use [M,I]=max(A,[],3) to get the indices with the highest values along the 3rd axis. M and I are therefore matrices of size n,m.

I want to get the values of B at the indices given in I. The result should be another n,m matrix.

What is the best way to get these values, without using for loops over n and m?

Thank you for you time :-)

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

KSSV 2016-11-3

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/310346-how-do-i-use-the-indices-from-m-i-max-a-3-to-find-the-corresponding-values-in-another-3d-array#answer_241790

在 MATLAB Online 中打开

clc; clear all ;
n = 10 ; m = 7 ; k = 10 ;
A = rand(n,m,k) ; 
B = rand(n,m,k) ;
[M,I]=max(A,[],3) ;
% using loop
iwant1 = zeros(size(M)) ;
for i = 1:n
    for j = 1:m
        iwant1(i,j) = B(i,j,I(i,j)) ;
    end
end
% without using loop 
B2D = reshape(B,n*m,k) ;
B2D = B2D(:,I(:)) ;
B2D = diag(B2D) ;
iwant2 = reshape(B2D,n,m) ;