在Numpy中，如何将两个二维数组进行压缩？

Question

例如，我有两个数组。

a = array([[0, 1, 2, 3],
           [4, 5, 6, 7]])
b = array([[0, 1, 2, 3],
           [4, 5, 6, 7]])

我该如何使用zip函数对a和b进行压缩？

c = array([[(0,0), (1,1), (2,2), (3,3)],
           [(4,4), (5,5), (6,6), (7,7)]])

?

- LWZ

Answer 1

您可以使用 dstack：

>>> np.dstack((a,b))
array([[[0, 0],
        [1, 1],
        [2, 2],
        [3, 3]],

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

如果必须使用元组：

>>> np.array(zip(a.ravel(),b.ravel()), dtype=('i4,i4')).reshape(a.shape)
array([[(0, 0), (1, 1), (2, 2), (3, 3)],
       [(4, 4), (5, 5), (6, 6), (7, 7)]],
      dtype=[('f0', '<i4'), ('f1', '<i4')])

对于Python 3+，您需要扩展zip迭代器对象。请注意，这种方法非常低效：

>>> np.array(list(zip(a.ravel(),b.ravel())), dtype=('i4,i4')).reshape(a.shape)
array([[(0, 0), (1, 1), (2, 2), (3, 3)],
       [(4, 4), (5, 5), (6, 6), (7, 7)]],
      dtype=[('f0', '<i4'), ('f1', '<i4')])

- Daniel

---

第二个命令出现了 TypeError: a bytes-like object is required, not 'zip' - 为什么会这样？ - Make42

@Make42 在Python 3中，zip返回一个迭代器。请查看已编辑的答案。 - Daniel

（在Python 3中）我不需要使用.ravel()函数。 - equaeghe

Answer 2

np.array([zip(x,y) for x,y in zip(a,b)])

- Aert

Answer 3

您可以指定转置索引：

c = np.array([a,b]).transpose(1,2,0)

- Andrew