numpy - 根据范围内的值过滤numpy数组中的行

Question

让一个数组：

 a =np.array([[1,2],[3,-5],[6,-15],[10,7]])

要获得第二列元素高于 -6 的行，可以这样做

>>> a[a[:,1]>-6]
array([[ 1,  2],
   [ 3, -5],
   [10,  7]])

但是如何在 -6;3 之间获得第二个元素的行？我试过：

>>> a[3>a[:,1]>-6]

还有（这会引发错误）：

>>> np.ma.masked_inside(a,-6,3)

这使：

 masked_array(data =
 [[-- --]
  [-- --]
  [6 -15]
  [10 7]],
         mask =
  [[ True  True]
  [ True  True]
  [False False]
  [False False]],
   fill_value = 999999)

但结果不是太清楚

谢谢 jp

Answer 1

>>> a[ (-6<a[:,1]) & (a[:,1]<3) ]

array([[ 1,  2],
      [ 3, -5]])

Answer 2

将np.ma.masked_inside(a, -6, 3)创建一个MaskedArray对象，其中 -6 和 3 之间的值被屏蔽（即标记为无效）。换句话说，您正在过滤掉-6 和 3 之间的值。相反，您应该使用np.ma.masked_outside(a, -6, 3)：

>>> a = np.array([[1,2],[3,-5],[6,-15],[10,2]])
>>> np.ma.masked_outside(a,-6,3)
>>> masked_array(data =
 [[1 2]
 [3 -5]
 [-- --]
 [-- 2]],
             mask =
 [[False False]
 [False False]
 [ True  True]
 [ True False]],
       fill_value = 999999)

请注意，使用此函数，您将逐个元素地过滤掉整个数组，这不是您想要的。

迄今为止，另一种解决方案中给出的索引方法是最简单易懂的。