numpy中stack函数

numpy中的stack函数比较的绕，具体实现如下：

----------------------------------------------------

import numpy as np
a=[1,2,3]
b=[4,5,6]
default_ab=np.stack((a,b))
v_ab=np.vstack((a,b))
h_ab=np.hstack((a,b))
axis0_ab=np.stack((a,b),axis=0)
axis1_ab=np.stack((a,b),axis=1)
print('default_ab:\n',default_ab)
print('v_ab:\n',v_ab)
print('h_ab:\n',h_ab)
print('axis0_ab:\n',axis0_ab)
print('axis1_ab:\n',axis1_ab)
temp=np.stack((axis1_ab),axis=1)
print('axis1_ab axis=1:\n',temp)

print('axis1_ab (axis=1)**2:\n',np.stack((temp),axis=1))

-------------------------------------------------------------

default_ab:
 [[1 2 3]
 [4 5 6]]
v_ab:
 [[1 2 3]
 [4 5 6]]
h_ab:
 [1 2 3 4 5 6]
axis0_ab:
 [[1 2 3]
 [4 5 6]]
axis1_ab:
 [[1 4]
 [2 5]
 [3 6]]
axis1_ab axis=1:
 [[1 2 3]
 [4 5 6]]
axis1_ab (axis=1)**2:
 [[1 4]
 [2 5]

 [3 6]]

-------------------------------------------------------

axis=1可以理解为转置

特别注意axis=1和hstack。