在3D矩阵中查找两个最大值的位置（MATLAB）

Question

我在3D矩阵（MATLAB）中识别两个最大值的位置时遇到了问题。假设我有矩阵A输出，如下所示：

A(:,:,1) =

     5     3     5
     0     1     0


A(:,:,2) =

     0     2     0
     8     0     8


A(:,:,3) =

     3     0     0
     0     7     7


A(:,:,4) =

     6     6     0
     4     0     0

对于第一个A（：，：，1），我想确定第一行的值最高（A=5）。但是我需要两个索引位置，在本例中，1和3。这与另一个A（：，：，：）相同。

我已经搜索了这么多，但由于我在MATLAB方面很差，我找不到解决这个问题的方法。

请一定要帮我。如果我不需要使用for循环来获得所需的输出，那就更好了。

Answer 1

以下代码提供了相应最大值的列和行。

第一步将获得包含第一和第二维度的每个子矩阵的最大值。由于max在默认情况下与第一维一起工作，矩阵被重塑以组合原始的第一维和第二维。

max_vals = max(reshape(A,size(A,1)*size(A,2),size(A,3)));
max_vals = 
           5     8     7     6

在第二步中，使用第三维上的arrayfun获得等于每个子矩阵的相应max_vals的元素索引。由于arrayfun的输出是单元格，因此使用cell2mat将输出转换为矩阵。最后一步，通过ind2sub将find中的线性索引转换为子索引。

[i,j] = ind2sub(size(A(:,:,1)),cell2mat(arrayfun(@(i)find(A(:,:,i)==max_vals(i)),1:size(A,3),'UniformOutput',false)))

i =
     1     2     2     1
     1     2     2     1

j =
     1     1     2     1
     3     3     3     2

因此，j中的值就是您想要的值。

Answer 2

快照#1查找每个3D切片的最大值索引-

%// Reshape A into a 2D matrix
A_2d = reshape(A,[],size(A,3))

%// Find linear indices of maximum numbers for each 3D slice
idx = find(reshape(bsxfun(@eq,A_2d,max(A_2d,[],1)),size(A)))

%// Convert those linear indices to dim1, dim2,dim3 indices and
%// present the final output as a Nx3 array
[dim1_idx,dim2_idx,dim3_idx] = ind2sub(size(A),idx)
out_idx_triplet = [dim1_idx dim2_idx dim3_idx]

样品运行-

>> A
A(:,:,1) =
     5     3     5
     0     1     0
A(:,:,2) =
     0     2     0
     8     0     8
A(:,:,3) =
     3     0     0
     0     7     7
A(:,:,4) =
     6     6     0
     4     0     0

out_idx_triplet =
     1     1     1
     1     3     1
     2     1     2
     2     3     2
     2     2     3
     2     3     3
     1     1     4
     1     2     4

out_idx_triplet（：，2）就是你要找的！

快照#2查找每个3D切片中最高两个数字的索引-

%// Get size of A
[m,n,r] = size(A)  

%// Reshape A into a 2D matrix
A_2d = reshape(A,[],r)

%// Find linear indices of highest two numbers for each 3D slice
[~,sorted_idx] = sort(A_2d,1,'descend')
idx = bsxfun(@plus,sorted_idx(1:2,:),[0:r-1]*m*n)

%// Convert those linear indices to dim1, dim2,dim3 indices
[dim1_idx,dim2_idx,dim3_idx] = ind2sub(size(A),idx(:))

%// Present the final output as a Nx3 array
out_idx_triplet = [dim1_idx dim2_idx dim3_idx]

out_idx_triplet（：，2）就是你要找的！