w Numpy, jak spakować dwie tablice 2D?

Question

Na przykład mam 2 tablice

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

Jak mogę zipa i b więc mogę

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

?

Answer 1

Można użyć dstack:

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

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

Jeśli musisz mieć krotki:

>>> 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')]) 

---

dla Pythona 3+ trzeba rozwinąć obiekt zip iteracyjnej. Należy pamiętać, że jest to strasznie niewydajne:

>>> 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')]) 

Answer 2

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