如何實現'內部'zip？

Question

兩個陣列x和y的形狀是(a,b)。我如何獲得組合形狀(a,b,2)？

我目前的解決方案是

z = np.zeros((a,b,2)) 
z[:,:,0] = x 
z[:,:,1] = y 

是否有可能實現這一目標而無需創建一個新的陣列？

Answer 1

您可以使用np.dstack：

In [2]: import numpy as np 
In [3]: a = np.random.normal(size=(4,6)) 
In [4]: b = np.random.normal(size=(4,6)) 

In [5]: np.dstack((a,b)).shape 
Out[5]: (4, 6, 2) 

和比較：

In [10]: d = np.dstack((a,b)) 
In [11]: c = np.zeros((4,6,2)) 
In [12]: c[:,:,0] = a 
In [13]: c[:,:,1] = b 

In [14]: np.allclose(c,d) 
Out[14]: True