función numpy.vsplit() | Python

numpy.vsplit()divide una array en múltiples sub-arrays verticalmente (en filas). vsplit es equivalente a dividir con eje = 0 (predeterminado), la array siempre se divide a lo largo del primer eje, independientemente de la dimensión de la array.

Sintaxis: numpy.vsplit(array, índices_o_secciones)
Parámetros: array
: [ndarray] Array que se dividirá en subarrays.
indices_or_sections : [array int o 1-D] Si indices_or_sections es un número entero, N, la array se dividirá en N arrays iguales a lo largo del eje.
Si indices_or_sections es una array 1-D de enteros ordenados, las entradas indican dónde se divide la array a lo largo del eje
Retorno: [ndarray] Una lista de sub-arrays.

Código #1:

# Python program explaining
# numpy.vsplit() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(9.0).reshape(3, 3)
  
gfg = geek.vsplit(arr, 1)
  
print (gfg)

Producción :

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

 
Código #2:

# Python program explaining
# numpy.vsplit() function
  
# importing numpy as geek 
import numpy as geek
  
arr = geek.arange(36.0).reshape(6, 6)
  
gfg = geek.vsplit(arr, 2)
  
print (gfg)

Producción :

[array([[  0.,   1.,   2.,   3.,   4.,   5.],
       [  6.,   7.,   8.,   9.,  10.,  11.],
       [ 12.,  13.,  14.,  15.,  16.,  17.]]), array([[ 18.,  19.,  20.,  21.,  22.,  23.],
       [ 24.,  25.,  26.,  27.,  28.,  29.],
       [ 30.,  31.,  32.,  33.,  34.,  35.]])]