matlab - 沿三维使用查找

Question

我有一个 2*3*3 矩阵a:

a=[1 1 1;1 1 1];
a(:,:,2)=[1 1 1;1 1 1]+1;
a(:,:,3)=[1 1 1;1 1 1]+2;

将三维视为深度值。我想找到 a 大于 2 的深度。在这种情况下，结果应该是:

[3 3 3;3 3 3]

有没有办法以矢量化的方式做到这一点？

我的尝试

我尝试了以下方法，但效果不佳:

inds=find(a>2) %find indices of elements>2  
lev=ceil(inds/2/3) %divide by the size of each layer. it returns the layer indices on which the an element>2 is

depths = reshape(lev,2,3) 

inds =

    13
    14
    15
    16
    17
    18


lev =

     3
     3
     3
     3
     3
     3

depths =

     3     3     3
     3     3     3

在这种情况下可行，但这是一个幸运的情况。

如果我使用:

a=[1 1 1;1 1 2];
a(:,:,2)=[1 1 1;1 1 2]+1;
a(:,:,3)=[1 1 1;1 1 2]+2;

现在它无法工作，因为在最后一列中我有多个大于 2 的值。确实:

inds=find(a>2) %find indices of elements>2  
lev=ceil(inds/2/3) %divide by the size of each layer. it returns the layer indices on which the an element>2 is

depths = reshape(lev,2,3)

inds =

    12
    13
    14
    15
    16
    17
    18


lev =

     2
     3
     3
     3
     3
     3
     3

inds =

    12
    13
    14
    15
    16
    17
    18


lev =

     2
     3
     3
     3
     3
     3
     3

Error using reshape To RESHAPE the number of elements must not change.

我看不到解决方法。

我还尝试将 find 与 'first' 选项一起使用，但完全没有成功。

Answer 1

您可以使用 max 函数的第二个输出:

[~,idx]=max(a>2,[],3);