例如我有2个数组
a = array([[0, 1, 2, 3], [4, 5, 6, 7]]) b = array([[0, 1, 2, 3], [4, 5, 6, 7]])
我怎么能zip a和b这样我得到
zip
a
b
c = array([[(0,0), (1,1), (2,2), (3,3)], [(4,4), (5,5), (6,6), (7,7)]])
?
您可以使用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+,您需要扩展zipiterator对象。请注意,这是非常低效的:
>>> 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')])