NumPy numpy.split 函数

• 

  简述

  此函数将数组沿指定轴划分为子数组。该函数接受三个参数。

  
  numpy.split(ary, indices_or_sections, axis)
  

  参数列表

  序号	参数及说明
  1	ary 要拆分的输入数组
  2	indices_or_sections 可以是整数，表示要从输入数组创建的大小相等的子数组的数量。如果此参数是一维数组，则条目指示要创建新子数组的点。
  3	axis 默认为 0

• 

  例子

  
  import numpy as np 
  a = np.arange(9) 
  print 'First array:' 
  print a 
  print '\n'  
  print 'Split the array in 3 equal-sized subarrays:' 
  b = np.split(a,3) 
  print b 
  print '\n'  
  print 'Split the array at positions indicated in 1-D array:' 
  b = np.split(a,[4,7])
  print b 
  

  它的输出如下 -

  
  First array:
  [0 1 2 3 4 5 6 7 8]
  Split the array in 3 equal-sized subarrays:
  [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]
  Split the array at positions indicated in 1-D array:
  [array([0, 1, 2, 3]), array([4, 5, 6]), array([7, 8])]