Python 按单个索引切片多行

我在numpy中有以下切片问题

a = np.arange(36).reshape(-1,4)
a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19],
       [20, 21, 22, 23],
       [24, 25, 26, 27],
       [28, 29, 30, 31],
       [32, 33, 34, 35]])

在我的问题中，在我的例子中，三行代表一个样本

如果我使用[0:2]获取以下内容，我希望以某种方式访问此矩阵：

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19],
       [20, 21, 22, 23]]

这是前两个坐标样本。 我必须从数组中提取大量这些坐标集

谢谢

基于，我找到了下面的解决方案，这给了我想要的结果

def chunks(l, n, indices):
    return np.vstack([l[idx*n:idx*n+n] for idx in indices])
chunks(a,3,[0,2])
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [24, 25, 26, 27],
       [28, 29, 30, 31],
       [32, 33, 34, 35]])

---

可能此解决方案可以改进，并且有人不需要堆叠。

如果三行是一个样本，您可以重新调整数组以反映这一点，使用奇特的索引检索样本，然后撤消形状更改：

>>> a = a.reshape(-1, 3, 4)
>>> a[[0, 2]].reshape(-1, 4)
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [24, 25, 26, 27],
       [28, 29, 30, 31],
       [32, 33, 34, 35]])

---

一般来说，

a[i*3，j*3]