将 numpy 数组切片成组

Question

我们有以下 numpy 数组：

A = [[0,1,2,3],
[2,3,4,5],
[1,2,3,4],
[1,1,1,1],
[2,2,2,2],
[3,3,3,3],
[0,1,2,3],
[2,3,4,5],
[1,2,3,4],
 [10,10,10,10],
[20,20,20,20],
[30,30,30,30]]

我想创建两个新的 arrays：

B = [[0,1,2,3],
[2,3,4,5],
[1,2,3,4],
[0,1,2,3],
[2,3,4,5],
[1,2,3,4]]

和

C = [[1,1,1,1],
[2,2,2,2],
[3,3,3,3],
 [10,10,10,10],
[20,20,20,20],
[30,30,30,30]]

基本上我想将数组 A 拆分为两个新的 arrays ，其中 B 采用 3 行组，数组 C 采用下一组 3 行。

Answer 1

可能有许多不同的方法可以解决这个问题，这里有一个：

import numpy as np
A = np.array([[0,1,2,3],
[2,3,4,5],
[1,2,3,4],
[1,1,1,1],
[2,2,2,2],
[3,3,3,3],
[0,1,2,3],
[2,3,4,5],
[1,2,3,4],
 [10,10,10,10],
[20,20,20,20],
[30,30,30,30]])

indices = np.reshape(np.arange(A.shape[0]),(-1,3))

B = A[indices[::2].flatten()]
C = A[indices[1::2].flatten()]

Answer 2

我们试试看：

tmp = A.reshape(-1, 3, A.shape[1])
B = tmp[::2]
C = tmp[1::2]