numpy.stack 与 numpy.concatenate 用法

在功能上 numpy.stack 与 numpy.concatenate 很相似，在使用上时常常搞混，两都之间最重要的一个区别时stack会增加一个维度，而concatenate不会增加维度，只是简单地完成拼接

 

 

以下数组为例

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


b = np.array(  [ [ 5,6  ],
                 [7, 8 ]] )

concatenate((a,b), 0) 数组拼接过程如下所示：

 

concatenate((a,b), 1) 

 

 

np.stack((a,b), 1)