如何 zip 2D arrays

Question

a=np.array([[1,2,3],[4,5,6],[7,8,9]])
b=np.array([[1,2,3],[4,5,6],[7,8,9]])

我有 2 个相同的 2D arrays，我正在尝试按元素对它们进行 zip。 它应该看起来像：

[[(1,1) (2,2), (3,3)]
[(4,4) (5,5) (6,6)]
[(7,7) (8,8) (9,9)]]

我试过下面的方法，但没有成功。 先把arrays、zip他们压平，转成列表，再转成数组再整形。

np.array(list(zip(np.ndarray.flatten(a),np.ndarray.flatten(b)))).reshape(a.shape)

我收到以下错误

cannot reshape array of size 18 into shape (3,3)

它没有将最终数组的元素(1,1) (2,2)等视为元组，而是作为单个元素。 因此，有 18 个元素。

这个问题已发布一次，但我没有找到适合我的答案。

Answer 1

不要zip ，使用 numpy 本机功能！ 你想要一个dstack ：

out = np.dstack([a, b])

output：

array([[[1, 1],
        [2, 2],
        [3, 3]],

       [[4, 4],
        [5, 5],
        [6, 6]],

       [[7, 7],
        [8, 8],
        [9, 9]]])