在Numpy中为3d数组制作2d掩码

Question

我有一个3D numpy数组after :all do @capybara_session.driver.quit() end 我想为它创建一个掩码。对于第一个阵列，它将是条件242> 122 + 123。有没有办法用numpy做条件？像[[[242,122,123],[111,30,12]]]这样的东西，其中a，b，c是数组中的值。

Answer 1

这只是猜测（关于你想要的）

In [134]: M=np.array([[[242,122,123],[111,30,12]]])
In [135]: M.shape
Out[135]: (1, 2, 3)
In [136]: M[:,:,0]>(M[:,:,1]+M[:,:,2])
Out[136]: array([[False,  True]], dtype=bool)
In [137]: M[_]
Out[137]: array([[111,  30,  12]])