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

Question

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

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

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