numpy - 「内側」のジッパーを実現するにはどうすればよいですか?

Question

2 つの配列の形状は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