如何將兩個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))