Python numpy.vsplit()函数

numpy.vsplit()函数将一个数组垂直（行）分割成多个子数组。vsplit等同于axis=0（默认）的split，无论数组维度如何，数组总是沿第一轴分割。

语法: numpy.vsplit(arr, indices_or_sections)
参数 :
arr : [ndarray] 阵列将被划分为子阵列。
indices_or_sections : [int or 1-D array] 如果indices_or_sections是一个整数，N，数组将沿轴线被分成N个相等的数组。
如果indices_or_sections是一个排序的整数的一维数组，那么这些条目表示该数组沿轴线的分割位置
返回： [ndarray] 一个子数的列表。

代码#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)

输出 :

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

代码#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)

输出 :

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