在Python中附加两个3d numpy数组

Question

我在想如何在python中附加两个3d numpy数组？

例如，我有一个形状为(81,81,61)，我想通过将原始数组附加到z方向上的自身来获得一个(81,81,122)形状数组。

Answer 1

的一种方法是使用其中np.dstack符连接沿所述第三轴线（d为深度）的阵列。

例如：

>>> a = np.arange(8).reshape(2,2,2) 
>>> np.dstack((a, a)) 
array([[[0, 1, 0, 1], 
     [2, 3, 2, 3]], 

     [[4, 5, 4, 5], 
     [6, 7, 6, 7]]]) 

>>> np.dstack((a, a)).shape 
(2, 2, 4) 

你也可以使用np.concatenate((a, a), axis=2)。