如何使用Numpy的瓷磚功能要得到這個結果

Question

我想廣播數組給予fashion.I現在瓷磚（）功能在numpy可用於廣播我試過了，但無法生成所需的輸出。

input=[ [1,2], 
     [3,4], 
     [4,5] ] #shape(3X2) 
numpy.tile(input,----) 
out put= [ [ [1,2], 
      [1,2] 
      ], 
      [ [3,4], 
      [3,4] 
      ], 
      [ [4,5], 
      [4,5], 
      ] 
     ] #shape(3,2,2) 

Answer 1

一種方法與np.repeat -

np.repeat(a,2,axis=0).reshape((a.shape) + (2,)) 

另外一個與np.repeat -

np.repeat(a[:,None],2,axis=1) # Or use np.newaxis in place of None 

隨着np.tile -

np.tile(a,2).reshape((a.shape) + (2,)) 

Answer 2

另一種選擇我s到堆棧input與自身和transpose

np.stack([input] * 2).transpose(1, 0, 2) 

array([[[1, 2], 
     [1, 2]], 

     [[3, 4], 
     [3, 4]], 

     [[4, 5], 
     [4, 5]]])