如何在python中选择2d数组的索引？

Question

我有一个numpy2d数组，其中笛卡尔坐标存储如下

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

如何将所有数组索引选择到另一个数组（y cordinate>；5)?

提前谢谢 普拉桑纳

Answer 1

为此，你可以很容易地使用列表理解。我建议使用下面的代码遍历列表“a”，并生成一个新的列表“b”，其中所有y值都大于5，如“if x[1]>；5". 希望这有帮助

  a = [[0,1],[1,1],[2,1],[4,6],[10,3], [2,10]]
  b = [x for x in a if x[1]>5]

现在，b=[[4，6]，[2，10]]

Answer 2

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