首頁 > 後端開發 > Python教學 > 如何使用自訂步幅高效建立 NumPy 子數組?

如何使用自訂步幅高效建立 NumPy 子數組?

Barbara Streisand
發布: 2024-11-30 22:18:12
原創
302 人瀏覽過

How to Efficiently Create NumPy Subarrays with Custom Strides?

具有自訂步長的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中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板