在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)。