在Python中将两个3D numpy数组连接起来

Question

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

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

- aregak

---

这个问题是第一个，所以我不想将其标记为重复，但有另一个相同的问题，它有更多的投票和稍微更好的答案（也来自Alex Riley）：https://dev59.com/VlsX5IYBdhLWcg3wALba - craq

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

- Alex Riley