python - 通过单个索引切片多行

Question

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

在我的问题中，总是三行代表一个样本，在我的例子中是坐标。

我想以某种方式访问​​这个矩阵，如果我使用 a[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]]

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

谢谢

基于How do you split a list into evenly sized chunks? ，我找到了以下解决方案，它给了我想要的结果。

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

也许这个解决方案可以改进，并且有人不需要堆叠。

Answer 1

如果三行是样本，您可以重新调整数组的形状以反射(reflect)这一点，使用花哨的索引来检索样本，然后撤消形状更改:

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