如何对numpy数组进行切片，使每个切片成为一个新数组

Question

说有这个数组：

x=np.arange(10).reshape(5,2)
x=array([[0, 1],
       [2, 3],
       [4, 5],
       [6, 7],
       [8, 9]])

是否可以对数组进行切片，使得除了第一行之外的每一行都将这些行组合成一个新矩阵？

例如：

newMatrixArray[0]=array([[0, 1],
       [2, 3]])

newMatrixArray[1]=array([[0, 1],
       [4, 5]])

newMatrixArray[2]=array([[0, 1],
       [6, 7]])

使用 for 循环很容易做到这一点，但是有没有一种 Python 的方式来做到这一点？

Answer 1

我们可以形成所有这些行索引，然后简单地索引到x。

因此，一种解决方案是 -

n = x.shape[0]
idx = np.c_[np.zeros((n-1,1),dtype=int), np.arange(1,n)]
out = x[idx]

样本输入、输出-

In [41]: x
Out[41]: 
array([[0, 1],
       [2, 3],
       [4, 5],
       [6, 7],
       [8, 9]])

In [42]: out
Out[42]: 
array([[[0, 1],
        [2, 3]],

       [[0, 1],
        [4, 5]],

       [[0, 1],
        [6, 7]],

       [[0, 1],
        [8, 9]]])

---

还有多种其他方法可以获取这些索引idx。让我们建议几个只是为了好玩。

一个broadcasting -

(np.arange(n-1)[:,None] + [0,1])*[0,1]

一个array-initialization -

idx = np.zeros((n-1,2),dtype=int)
idx[:,1] = np.arange(1,n)

一个cumsum -

np.repeat(np.arange(2)[None],n-1,axis=0).cumsum(0)

一个带有列表扩展 -

np.c_[[[0]]*(n-1), range(1,n)]

此外，为了提高性能，请使用 np.column_stack 或 np.concatenate 代替 np.c_。

Answer 2

我的方法是列表理解：

np.array([
    [x[0], x[i]]
    for i in range(1, len(x))
])