如何使用另一個數組中的索引來存取 NumPy 中的數組元素？

在NumPy 中使用數組索引存取數組元素

NumPy 的索引函數提供了強大的資料操作技術，包括從一個數組中選擇元素的能力使用另一個陣列指定的索引。實現此目的：

方法1：使用高階索引

A[np.arange(A.shape[0])[:,None] ,B]<br>

此程式碼利用進階索引，其中np.arange(A.shape[0])[:,None] 建立一個列向量，其中的每一行都有索引A. 將其與 B 結合，可以沿行和列對 A 進行索引。

方法2：線性索引

m,n = A.shape<br>np.take(A,B n*np.arange(m )[:,None])<br>

此方法利用線性索引，其中 A中的每個元素都被尋址通過單一索引。它首先透過將 B 中的對應行新增至使用 np.arange 產生的線性序列來計算線性索引。然後使用該線性索引從 A 檢索元素。

範例用法：

給定矩陣A：

與索引矩陣B :

應用🎜>應用這些方法會產生預期的結果：

以上是如何使用另一個數組中的索引來存取 NumPy 中的數組元素？的詳細內容。更多資訊請關注PHP中文網其他相關文章！