numpy数组的分割与合并

合并

np.newaxis

import numpy as np

a=np.array([1,2,3])[:,np.newaxis]#变成列向量

b=np.array([4,5,6])[:,np.newaxis]#变成列向量

c=np.vstack((a,b)) #vertical  stack

d=np.hstack((a,b))#horizontal stack

print(a.shape ,b.shape)   # ((3,1),(3,1))

print(c.shape)    # (6,1)

print(d.shape)    # (3,2)

np.concatenate

import numpy as np

a=np.array([1,2,3])[:,np.newaxis]

b=np.array([4,5,6])[:,np.newaxis]

c=np.concatenate((a,b,b,a))

print(c)#按列合并

d=np.concatenate((a,b,b,a),axis=0)#设置在列上合并

print(d)

e=np.concatenate((a,b,b,a),axis=1)#设置在行上合并，3行4列

print(e)

分割

import numpy as np

a=np.arange(12).reshape((3,4))

print(a)

print(np.split(a,2,axis=1)) #   vertical纵向分2部分

print(np.split(a,3,axis=0)) #   horizontal横向分3部分

print(np.array_split(a,3,axis=1))#纵向分3部分

print(np.vsplit(a,3)) #横着分3组

print(np.hsplit(a,2)) #竖着分两组