提取Nx3陣列的特定行，從而每列滿足條件

Question

假設我有一個Nx3的數組，我想要滿足的元素說：

4 < col1 < 13, 5 > col2 >  3, 10 > col3 > 6

即應用於：

1,2,3
4,5,6
9,4,7

然后它將提供一個Mx3數組，它只包含滿足所有三個條件的行。 對於上面的例子，它輸出

9,4,7

我想過做一個循環，但我認為numpy或類似的東西必須有更快的東西（例如np.where？）

Answer 1

您為每列構建一個布爾數組：

cond1 = (my_array[:, 0] > 4) & (my_array[:, 0] < 13)
cond2 = (my_array[:, 1] > 3) & (my_array[:, 1] < 5)
cond3 = (my_array[:, 2] > 6) & (my_array[:, 2] < 10)

然后你and你的三個條件在一起：

cond = cond1 & cond2 & cond3

最后你用它來從你的數組中選擇行：

out = my_array[cond]