in Numpy, wie zwei 2-D-Arrays zip?

Question

Zum Beispiel Ich habe 2 Arrays

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

Wie kann ich zipa und b so erhalte ich

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

?

Answer 1

können Sie dstack verwenden:

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

---

Für Python 3+ Sie das zip Objekt Iterator erweitern müssen:

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

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

Wenn Sie Tupel haben müssen. Bitte beachten Sie, dass dies schrecklich ineffizient:

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