数组的分隔

import  numpy
x=numpy.arange(9)

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

#对一维数组进行分隔
a=numpy.split(x,3)   #将数组平均分成3份
#传递数组按位置进行分隔
b=numpy.split(x,[3,5])
print(a)
print(b)

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

#对二维数组进行分隔
c=numpy.array([[1,2,3,4],[4,3,2,1],[5,6,8,4],[5,7,9,4]])
w,r=numpy.split(c,2,axis=0)
#按位置进行分隔
t,e,v=numpy.split(c,[2,3],axis=0)

　print(w)
  print(r)

  print(t)
  print(e)
  print(v)

[[1 2 3 4]                    
[4 3 2 1]]
[[5 6 8 4]
[5 7 9 4]]

 

[[1 2 3 4]
[4 3 2 1]]
[[5 6 8 4]]
[[5 7 9 4]]

 

ti,wi=numpy.split(c,2,axis=1)   #平均分割

#按数组进行分隔
vi,wu,di=numpy.split(c,[2,3],axis=1)
#print(vi)
#print(wu)
#print(di)

#使用hsplit进行水平方向分隔
q1,q2=numpy.hsplit(c,2)
#print(q1)
#print(q2)

#按位置进行分隔
a1,a2,a3=numpy.hsplit(c,[2,3])

#使用vsplit进行垂直分隔
b1,b2,b3=numpy.vsplit(c,[2,3])