具有自訂步長的Numpy 子數組
可以透過多種方式從具有給定步長的NumPy 數組建立子數組。這裡有兩種有效的方法:
廣播方法:
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)]
跨步方法:
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))
範例:
考慮NumPy陣列a:
a = numpy.array([1,2,3,4,5,6,7,8,9,10,11])
建立長度為5 的子數組步幅為3,我們可以使用方法:
broadcasting_result = broadcasting_app(a, L=5, S=3) strided_result = strided_app(a, L=5, S=3) print(broadcasting_result) >> [[ 1 2 3 4 5] [ 4 5 6 7 8] [ 7 8 9 10 11]] print(strided_result) >> [[ 1 2 3 4 5] [ 4 5 6 7 8] [ 7 8 9 10 11]]
兩種方法都能有效地產生所需的子數組矩陣。
以上是如何使用自訂步幅高效建立 NumPy 子數組?的詳細內容。更多資訊請關注PHP中文網其他相關文章!