在Matlab中找到固体物体的顶面

Question

我在名为mask (1000x1000x1000 uint16) 的 Matlab 3D 数组中对实体对象进行了二进制分割。我需要提取顶面，即每个列向量的第一个非零元素（在第二维和第三维上）。我可以用下面的代码 sn-p 做到这一点：

s = zeros(1000,1000); 
for y=1:1000
    for x=1:1000
        s(x,y) = find(mask(:,x,y),1);
    end 
end

有没有办法向量化代码并避免for 循环？

Answer 1

查找maximum 值的索引（在您的情况下为一个）。 Squeeze 这些索引用于删除单一维度并转换为所需的形式 (s)。

[~, ind] = max(mask); 
s = squeeze(ind);