在Numpy中，如何压缩两个2-D数组？
在Numpy中，如何压缩两个2-D数组？
提问于 2013-07-31 10:07:34
例如，我有两个数组
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)]])回答 3
Stack Overflow用户
回答已采纳
发布于 2013-07-31 10:29:19
您可以使用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迭代器，需要展开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')])Stack Overflow用户
发布于 2013-07-31 11:22:42
np.array([zip(x,y) for x,y in zip(a,b)])Stack Overflow用户
发布于 2021-03-10 10:03:14
您还可以指定转置索引：
c = np.array([a,b]).transpose(1,2,0)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/17960441
复制相关文章
相似问题
腾讯云开发者
