python - 如何在 python 中选择二维数组的索引？

Question

我有一个 numpy 二维数组，其中笛卡尔坐标存储如下。

a = [[0,1],[1,1],[2,1],[4,6],[10,3], [2,10]]

如何将所有数组索引选择到另一个数组(y坐标> 5)？

提前致谢普拉萨那

Answer 1

>>> a = np.array([[0,1],[1,1],[2,1],[4,6],[10,3], [2,10]])
>>> a[:, 1] > 5 # Boolean mask
array([False, False, False,  True, False,  True])
>>> a[a[:, 1] > 5] # Filtered array
array([[ 4,  6],
       [ 2, 10]])
>>> np.where(a[:, 1] > 5)[0] # Array of indices
array([3, 5])