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)) #豎着分兩組