使用 numpy 有效地找到最低的非屏蔽点

Question

这里的应用程序正在寻找“云基础”，但原则适用于任何地方。我有一个 numpy 屏蔽 3-D 数组（我们会说它对应于尺寸为 z、y、x 的 3-D 网格框），其中我已经屏蔽了值小于 0.1 的所有点。我想找到的是，在每个 x,y 点，未被屏蔽的最低 z 点索引（不是 z 中的最小值，最小 z 坐标）是多少。我可以想到一些简单的方法来做到这一点，例如：

for x points:
    for y points:
        minz=-1
        for z points:
            if x,y,z is not masked:
                 minz = z 
                 break

但是，这似乎真的很低效，我相信有一种更有效或更 Pythonic 的方式来做到这一点。我在这里错过了什么？

编辑：我不需要使用掩码数组，但这似乎是提出问题的最简单方法 - 我可以找到某个特定条件下的最低点不使用掩码数组的阈值。

编辑 2： 我正在寻找的想法（以 z=0 为最低点）：

input:
[[[0,1],
  [1,5]],

 [[3,3],
  [2,4]],

 [[2,1],
  [4,9]]]

threshold: val >=3
output:
[[1,1],
 [2,0]]

Answer 1

假设 A 作为输入数组，你可以这样做 -

np.where((A < thresh).all(0),-1,(A >= thresh).argmax(0))

样本运行

运行#1：

In [87]: A
Out[87]: 
array([[[0, 1],
        [1, 5]],

       [[3, 3],
        [2, 4]],

       [[2, 1],
        [4, 9]]])

In [88]: thresh = 3

In [89]: np.where((A < thresh).all(0),-1,(A >= thresh).argmax(0))
Out[89]: 
array([[1, 1],
       [2, 0]])

运行#2：

In [82]: A
Out[82]: 
array([[[17,  1,  2,  3],
        [ 5, 13, 11,  2],
        [ 9, 16, 11, 19],
        [11, 16,  6,  3],
        [15,  9, 14, 14]],

       [[18, 19,  5,  8],
        [13, 13, 17,  2],
        [17, 12, 16,  0],
        [19, 14, 12,  5],
        [ 7,  8,  4,  7]],

       [[10, 12, 11,  2],
        [10, 18,  6, 15],
        [ 4, 16,  0, 16],
        [16, 18,  2,  1],
        [10, 19,  9,  4]]])

In [83]: thresh = 10

In [84]: np.where((A < thresh).all(0),-1,(A >= thresh).argmax(0))
Out[84]: 
array([[ 0,  1,  2, -1],
       [ 1,  0,  0,  2],
       [ 1,  0,  0,  0],
       [ 0,  0,  1, -1],
       [ 0,  2,  0,  0]])