Dividir un arreglo de numpy en grupos.
Tenemos el siguiente array de 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]]
Me gustaría crear dos nuevos 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]]
y
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]]
Básicamente, me gustaría dividir el array A en dos nuevos arrays donde B tome grupos de 3 filas y el array C tome los siguientes grupos de 3 filas.
davy.ai
Puede usar la función
numpy.split()
para lograr esto.