Python numpy.vsplit()用法及代码示例

numpy.vsplit()函数将一个数组垂直(行)拆分为多个子数组。 vsplit等效于使用axis = 0进行拆分(默认)，无论数组尺寸如何，数组始终沿第一个轴拆分。

用法： numpy.vsplit(arr, indices_or_sections)

参数：

arr  :[ndarray] Array to be divided into sub-arrays.

indices_or_sections  :[int or 1-D array] If indices_or_sections is an integer, N, the array will be divided into N equal arrays along axis.

If indices_or_sections is a 1-D array of sorted integers, the entries indicate where along axis the array is split

Return :[ndarray] A list of sub-arrays.

代码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.]])]