根据每个数组的特定索引中的值，在 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]