与えられたストライド/ stepsizeでnumpy配列からサブ配列を取る

Question

私はPython Numpy配列配列を持っているといいますか？次のように私は結果行列3.ストライドで長さ5のこの配列からサブシーケンスのマトリックスを作成したい

numpy.array([1,2,3,4,5,6,7,8,9,10,11].) 

は、それゆえになります。

これを実現するための1つの可能な方法は次のようになり

numpy.array([[1,2,3,4,5],[4,5,6,7,8],[7,8,9,10,11]]) 

をforループを使用します。

result_matrix = np.zeros((3, 5)) 
for i in range(0, len(a), 3): 
    result_matrix[i] = a[i:i+5] 

これを実装するクリーナーの方法はNumpyですか。

Answer 1

アプローチ＃1：broadcasting使用 -

を

def broadcasting_app(a, L, S): # Window len = L, Stride len/stepsize = S 
    nrows = ((a.size-L)//S)+1 
    return a[S*np.arange(nrows)[:,None] + np.arange(L)] 

アプローチ＃2：より効率的な使用NumPy strides -

def strided_app(a, L, S): # Window len = L, Stride len/stepsize = S 
    nrows = ((a.size-L)//S)+1 
    n = a.strides[0] 
    return np.lib.stride_tricks.as_strided(a, shape=(nrows,L), strides=(S*n,n)) 

サンプル実行 -

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

In [144]: broadcasting_app(a, L = 5, S = 3) 
Out[144]: 
array([[ 1, 2, 3, 4, 5], 
     [ 4, 5, 6, 7, 8], 
     [ 7, 8, 9, 10, 11]]) 

In [145]: strided_app(a, L = 5, S = 3) 
Out[145]: 
array([[ 1, 2, 3, 4, 5], 
     [ 4, 5, 6, 7, 8], 
     [ 7, 8, 9, 10, 11]])