製作3D陣列與1D陣列，這取決於一維數組的索引零填充numpythonically

Question

使用

n=4 
cols=3 
k=np.vstack((a,b,a*b,np.zeros((n,cols)),a,b,a,a,b**2,np.zeros((n,cols)),a*2,a)).T.reshape(-1,2,n+5) 

，其中一個考慮下面的一維數組

a=np.arange(3)+9 
b=np.arange(3)+5 

目前我正在初始化新的3D陣列和b將始終是相同的形狀

其結果在

array([[[ 9., 5., 45., 0., 0., 0., 0., 9., 5.], 
     [ 9., 9., 25., 0., 0., 0., 0., 18., 9.]], 

     [[ 10., 6., 60., 0., 0., 0., 0., 10., 6.], 
     [ 10., 10., 36., 0., 0., 0., 0., 20., 10.]], 

     [[ 11., 7., 77., 0., 0., 0., 0., 11., 7.], 
     [ 11., 11., 49., 0., 0., 0., 0., 22., 11.]]]) 

我將如何使用類似的技術，也沒有一個for循環中，改變零填充到以下：

array([[[ 9., 5., 45., 9., 5., 0., 0., 0., 0.], 
     [ 9., 9., 25., 18., 9., 0., 0., 0., 0.]], 

     [[ 10., 6., 60., 0., 0., 10., 6., 0., 0.], 
     [ 10., 10., 36., 0., 0., 20., 10., 0., 0.]], 

     [[ 11., 7., 77., 0., 0., 0., 0., 11., 7.], 
     [ 11., 11., 49., 0., 0., 0., 0., 22., 11.]]]) 

Answer 1

人們可以使用advanced-indexing那些數組值指定到零初始化數組給定的列索引 -

out = np.zeros((3,2,9),dtype=bool) 
vals = np.array([[a,b,a*b,a,b],[a,a,b**2,2*a,a]]) 
out[np.arange(3)[:,None],:, idx] = vals.T 

樣品運行 -

In [448]: a 
Out[448]: array([ 9, 10, 11]) 

In [449]: b 
Out[449]: array([5, 6, 7]) 

In [450]: out 
Out[450]: 
array([[[ 9., 5., 45., 9., 5., 0., 0., 0., 0.], 
     [ 9., 9., 25., 18., 9., 0., 0., 0., 0.]], 

     [[ 10., 6., 60., 0., 0., 10., 6., 0., 0.], 
     [ 10., 10., 36., 0., 0., 20., 10., 0., 0.]], 

     [[ 11., 7., 77., 0., 0., 0., 0., 11., 7.], 
     [ 11., 11., 49., 0., 0., 0., 0., 22., 11.]]])