numpy reshape 2次元行列から3次元行列を積み重ねることによって

Question

私は形状がnum32の行列を持っています(x,y)。私は、形状が(x,1)のyベクトルの垂直スタックであるテンソルを取得したいと思います。私はnp.reshape(2,3,1)を行う際

array([[ 1., 2.], 
     [ 1., 2.], 
     [ 1., 2.]]) 

： は、私は次の行列を考えてみましょう。私が取得します：

array([[[ 1.], 
     [ 2.], 
     [ 1.]], 
     [[ 2.], 
     [ 1.], 
     [ 2.]]]) 

が、私はこれをしたい：あなたは形状（2,3,1）をしたい

array([[[ 1.], 
     [ 1.], 
     [ 1.]], 
     [[ 2.], 
     [ 2.], 
     [ 2.]]]) 

Answer 1

In [135]: arr=np.array([[1,2],[1,2],[1,2]]) 
In [136]: arr.shape 
Out[136]: (3, 2) 
In [137]: arr.transpose() 
Out[137]: 
array([[1, 1, 1], 
     [2, 2, 2]]) 
In [138]: arr.transpose()[:,:,None] 
Out[138]: 
array([[[1], 
     [1], 
     [1]], 

     [[2], 
     [2], 
     [2]]]) 

。 （3,2）から始めると、2軸を切り替えて1軸を追加する必要があることを意味します。これはどちらの順序でも行うことができます。ここでは、切り替えを行うために転置を選択し、次元を追加するには[:,:,None]を選択します。

arr.reshape(3,2,1).swapaxes(0,1)も有効です。より明白に、np.stack(arr[:,:,None],1)。