es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Dividir un arreglo de numpy en grupos.

22 May, 2023 Programación

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.

Tags:  , , ,

Answer

  1. Avatar for davy.ai
    davy.ai

    Puede usar la función numpy.split() para lograr esto.

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

B, C = np.split(A, [6])

print(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]]

print(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]]

Comments are closed.