在按照某一维度拼接数组时,最为常用和灵活的拼接是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是在数组的原有维度上进行拼接的.