返回多个numpy数组范围

Question

假设有如下数组：

a = np.array([[1,2],
              [2,3],
              [2,3],
              [2,3],
              [4,5],
              [3,4],
              [2,3]])

我将如何返回多行，这是我想要实现的（我知道语法错误，但这会让你知道我想要实现的目标）：

a[0:2 & 5:6,:]

我无法传递单个行，因为在我的实际代码中，我需要包含更大的范围，例如20:60＆amp; 90:160等等。

Answer 1

这有效：

>>> a[np.r_[:2, 5:6], :]
array([[1, 2],
       [2, 3],
       [3, 4]])

np.r_：

  

将切片对象转换为沿第一轴的连接。