NumPy numpy.vsplit 函数

• 

  简述

  numpy.vsplit 是 split() 函数的一个特例，其中轴为 1 表示垂直拆分，无论输入数组的维度如何。下面的例子清楚地说明了这一点。
• 

  例子

  
  import numpy as np 
  a = np.arange(16).reshape(4,4) 
  print 'First array:' 
  print a 
  print '\n'
  print 'Vertical splitting:' 
  b = np.vsplit(a,2) 
  print b
  

  复制

  它的输出如下 -

  
  First array:
  [[ 0 1 2 3]
   [ 4 5 6 7]
   [ 8 9 10 11]
   [12 13 14 15]]
  Vertical splitting:                                                           
  [array([[0, 1, 2, 3],                                                         
         [4, 5, 6, 7]]), array([[ 8,  9, 10, 11],                               
         [12, 13, 14, 15]])] 
  

  复制