numpy中在某一维度拼接数组 concatenate和stack的区别

在按照某一维度拼接数组时，最为常用和灵活的拼接是stack和concatenate

但它们的效果也有一定的区别

>>> import numpy as np
>>> A=np.ones((2,1,3))
>>> B=np.ones((2,1,3))

我们分别使用stack和concatenate来在第二维度上拼接A和B两个数组

>>> C_stack=np.stack((A,B),axis=1)
>>> C_stack.shape
(2, 2, 1, 3)

>>> C_con=np.concatenate((A,B),axis=1)
>>> C_con.shape
(2, 2, 3)

可以看到

对于stack函数而言,使其在某一维度上拼接就在该维度多出了一个维度,它在原有基础上创造了新的一个维度,而concatenate是在数组的原有维度上进行拼接的.