如何在 NumPy 中以特定步長有效提取子數組？

使用NumPy 使用跨步提取子數組

給定一個NumPy 數組，通常需要將其劃分為具有特定跨度的子數組。本問題探討如何使用廣播和 NumPy 步幅有效地實現此目的。

解 1：廣播

broadcasting_app 函式使用廣播來建構所需的子陣列。它根據數組的大小、子數組長度和步幅計算行數。然後，它使用廣播建立一個新數組，其中每一行代表一個子數組。

解決方案 2：NumPy Strides

strided_app 函數利用 NumPy 高效率的步幅處理機制。它像以前一樣計算行數和步幅。然後，它利用 as_strided 函數建立具有所需步幅和形狀的新陣列。

範例用法

為了說明這些解，請考慮以下Python 程式碼：

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
L = 5  # Subarray length
S = 3  # Stride

print(broadcasting_app(a, L, S))
print(strided_app(a, L, S))

輸出：

[[ 1  2  3  4  5]
 [ 4  5  6  7  8]
 [ 7  8  9 10 11]]
[[ 1  2  3  4  5]
 [ 4  5  6  7  8]
 [ 7  8  9 10 11]]

輸出：兩種方法有效地產生所需的子數組矩陣。

以上是如何在 NumPy 中以特定步長有效提取子數組？的詳細內容。更多資訊請關注PHP中文網其他相關文章！