快速numpy的方式重複（縱向）的二維陣列

Question

各佔一半。假設我們有以下的二維數組：

In [200]: a = np.arange(8).reshape(4,2) 

In [201]: a 
Out[201]: 
array([[0, 1], 
     [2, 3], 
     [4, 5], 
     [6, 7]]) 

如何可以重複的是各佔一半，所以我有以下二維數組：

array([[0, 1], 
     [2, 3], 
     [0, 1], 
     [2, 3], 
     [4, 5], # second half 
     [6, 7], 
     [4, 5], 
     [6, 7]]) 

我嘗試產生錯誤的結果：

In [202]: np.tile(np.split(a, 2), 2).reshape(-1,2) 
Out[202]: 
array([[0, 1], 
     [0, 1], 
     [2, 3], 
     [2, 3], 
     [4, 5], 
     [4, 5], 
     [6, 7], 
     [6, 7]]) 

Answer 1

重塑分裂第一軸分爲兩個給我們一個三維陣列，然後沿着第一重複，最終重塑回2D -

np.repeat(a.reshape(-1,2,2),2,axis=0).reshape(-1,2) 

歸納了 -

def repeat_blocks(a): 
    N = a.shape[0] 
    B = N//2 # Block length 
    R = 2 # number of repeats 
    out = np.repeat(a.reshape(N//B,B,-1),R,axis=0).reshape(N*R,-1) 
    return out 

樣品試驗 -

案例＃1：

In [120]: a 
Out[120]: 
array([[0, 1], 
     [2, 3], 
     [4, 5], 
     [6, 7]]) 

In [121]: repeat_blocks(a) 
Out[121]: 
array([[0, 1], 
     [2, 3], 
     [0, 1], 
     [2, 3], 
     [4, 5], 
     [6, 7], 
     [4, 5], 
     [6, 7]]) 

案例＃2：

In [123]: a 
Out[123]: 
array([[ 0, 1], 
     [ 2, 3], 
     [ 4, 5], 
     [ 6, 7], 
     [ 8, 9], 
     [10, 11]]) 

In [124]: repeat_blocks(a) 
Out[124]: 
array([[ 0, 1], 
     [ 2, 3], 
     [ 4, 5], 
     [ 0, 1], 
     [ 2, 3], 
     [ 4, 5], 
     [ 6, 7], 
     [ 8, 9], 
     [10, 11], 
     [ 6, 7], 
     [ 8, 9], 
     [10, 11]])