数组的分隔

• numpy.split 函数沿特定的轴将数组分割为子数组，格式如下：

• numpy.split(ary, indices_or_sections, axis)

  ary：被分割的数组

  indices_or_sections：如果是一个整数，就用该数平均切分，如果是一个数组，为沿轴切分的
  位置。
  axis：沿着哪个维度进行切向，默认为 0，横向切分。为 1 时，纵向切分。
• 代码

#导入numpy模块
import numpy as np  
#创建一维的数组
a = np.arange(1,13)
#调用split函数进行分隔
print('传递整数，平均分隔')
r = np.split(a,4,axis=0)
print(r)

print('传递数组，按位置分隔')
r = np.split(a,[4,6])
print(r)

#二维数组进行分隔
a = np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])
print(a)
print('axis=0 垂直方向 平均分隔')
r,w = np.split(a,2,axis=0)
print(r)
print(w)
print('axis=0 垂直方向 按位置分隔')
g,k,l = np.split(a,[2,3],axis=0)
print(g)
print(k)
print(l)

传递整数，平均分隔
[array([1, 2, 3]), array([4, 5, 6]), array([7, 8, 9]), array([10, 11, 12])]
传递数组，按位置分隔
[array([1, 2, 3, 4]), array([5, 6]), array([ 7,  8,  9, 10, 11, 12])]      
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]
 [13 14 15 16]]
axis=0 垂直方向 平均分隔
[[1 2 3 4]
 [5 6 7 8]]
[[ 9 10 11 12]
 [13 14 15 16]]
axis=0 垂直方向 按位置分隔
[[1 2 3 4]
 [5 6 7 8]]
[[ 9 10 11 12]]
[[13 14 15 16]]