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

Question

将两个numpy数组（aa, bb）与可能不同数量的子数组（M1不一定等于M2）但等长（{{1} }）和浮点数（N）

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) 的所有子数组中同时< e_max的所有元素的索引及其补码;即，{{1>}中<{1>}的元素的索引至少 aa, bb的一个子数组。

使用>= e_max进行此操作的有效方法是什么？

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)))