在功能上 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)