如何使用整數數組索引從 NumPy 數組中提取元素？

使用 NumPy 數組進行索引提取

在 NumPy 中處理多維數組時，通常需要根據索引提取特定元素。在這方面，可能會出現某些情況，其中一個陣列 (B) 充當另一個陣列 (A) 第二維的索引。

考慮一個具有兩個陣列A 和B 的範例：

A = np.array([[0,1], 
              [2,3], 
              [4,5]])
B = np.array([[1],
              [0],
              [1]], dtype='int')

我們的目標是從A 的每一行中提取一個元素，其中元素索引由B 中對應的元素確定。所需的輸出是：

C = np.array([[1],
              [2],
              [5]])

使用整數的解數組索引：

我們可以使用NumPy 的純整數陣列索引來取代廣播B：

A[np.arange(A.shape[0]),B.ravel()]

這裡， np.arange(A.shape[0]) 建立一個對應於A 中行數的索引數組，而 B.ravel() 將 B 展平為一維數組。透過組合這些，我們可以從 A 中提取適當的元素。

範例運行：

print(A)
print(B)
print(A[np.arange(A.shape[0]),B.ravel()])

輸出：

[[0 1]
 [2 3]
 [4 5]]
[[1]
 [0]
 [1]]
[1 2 5]

這種方法提供我們的提取問題了簡潔高效的解決方案。

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