numpy

• np.stack的用法

np.stack((a,b,c,...),axis=0)
np.stack 第一个参数为一个元组
第二个参数是维度
方法合并2个np.array，并增加一个维度

In [2]: import numpy as np                                                                                                                                     
In [3]: a = np.array([1,2])                                                                                                                                    
In [4]: b = np.array([3,4])                                                                                                                                    
In [5]: c = np.stack((a,b),axis=0)                                                                                                                             
In [6]: d = np.stack((a,b),axis=1)                                                                                                                             
In [7]: c                                                                                                                                                      
Out[7]: 
array([[1, 2],
       [3, 4]])
In [8]: d                                                                                                                                                      
Out[8]: 
array([[1, 3],
       [2, 4]])