Python | método numpy.array_split()

Con la ayuda del numpy.array_split()método, podemos obtener la array dividida que tiene diferentes dimensiones usando el numpy.array_split()método.

Sintaxis: numpy.array_split()

Return : Devuelve la array dividida de una dimensión.

Ejemplo n.º 1:
en este ejemplo, podemos ver que al usar el numpy.array_split()método, podemos dividir la array en el número de subarreglos pasándolo como parámetro.

# import numpy
import numpy as np
  
array = np.arange(9)
  
# using numpy.array_split() method
gfg = np.array_split(array, 4)
  
print(gfg)

Producción :

[array([0, 1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]

Ejemplo #2:

# import numpy
import numpy as np
  
array = [[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]]
  
# using numpy.array_split() method
gfg = np.array_split(array, 3)
  
print(gfg)

Producción :

[array([[1, 2, 3]]), array([[4, 5, 6]]), array([[7, 8, 9]])]