如何将两个1d numpy数组拉到2d numpy数组中

Question

我有2个numpy 1d数组，例如，

a = np.array([1,2,3,4,5]) 
b = np.array([6,7,8,9,10]) 

然后我怎样才能获得2D阵列[[1,6]，[2,7]，[3,8]，[4,9]，[5,10]

Answer 1

的答案就在你的问题：

np.array(list(zip(a,b))) 

编辑：

虽然我的POS t根据OP的要求给出答案，转换到列表并返回到NumPy数组需要一些开销。

因此，dstack将是一个计算有效的替代方案（参考@ zipa的答案）。在发布这个答案时，我并不知道dstack，所以在@zipa中介绍这篇文章的时候感谢。

Answer 2

您可以使用zip

np.array(list(zip(a,b))) 
array([[ 1, 6], 
    [ 2, 7], 
    [ 3, 8], 
    [ 4, 9], 
    [ 5, 10]]) 

Answer 3

如果你有numpy的阵列可以使用dstack()：

import numpy as np 

a = np.array([1,2,3,4,5]) 
b = np.array([6,7,8,9,10]) 

c = np.dstack((a,b)) 
#or 
c = np.column_stack((a,b))