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])