根據每個數組的特定索引中的值，在 2D numpy 數組中按索引查找和切片？

Question

我有很多 2D numpy arrays 看起來像這樣：

arr = np.array([[2,2],
                [2,3],
                [3,4],
                [3,5],
                [3,6],
                [4,7]))

如何查詢二維數組並檢索 0 索引中所有值為 3 的 arrays？ 所以我想要：

[[3,4],[3,5],[3,6]]

我考慮過將它變成列表列表，但這似乎效率低下，因為我有很多查詢要提出。 使用 np.argwhere 或 np.where 似乎也不按索引值進行隔離。

謝謝你。

Answer 1

對於高級索引，首先我們必須通過arr[:, 0] == 3找到具有指定編號作為其第一個參數的元素，因此：

arr[arr[:, 0] == 3]