返回多个范围的 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 & 90:160等

Answer 1

这行得通：

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

np.r_：

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