小于/大于numpy子数组中的float的元素的索引

Question

取两个numpy数组（ aa, bb ），其中子数组的数量可能不同（ M1不一定等于M2 ），但长度相等（ N ），而float（ e_max ）

M1, M2, N = np.random.randint(1, 3), np.random.randint(1, 3), 100
e_max = .3
aa = np.random.rand(M1, N)
bb = np.random.rand(M2, N)

我需要在aa, bb 所有子数组及其补数中同时< e_max的所有元素的索引； 即， 至少一个aa, bb子数组中>= e_max的那些元素的索引。

用numpy做到这一点的有效方法是什么？

Answer 1

IIUC有两种方法。

与any -

np.flatnonzero(~((aa>=e_max).any(0) | (bb>=e_max).any(0)))

与all -

np.flatnonzero(((aa<e_max).all(0) & (bb<e_max).all(0)))