如何使用整數數組索引根據另一個數組中的索引從一個數組中提取元素？

利用整數數組索引根據輔助數組索引提取元素

在給定的場景中，目標是從數組A 使用第二個數組B 中指定的索引。更直接的方法是使用NumPy 的整數陣列索引，而不是依賴np.take 或np.choose：

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

以下是此程式碼如何實現預期的結果：

• np.arange(A.shape[0]) 建立一個從0 到(A.shape[0] - 1) 的索引數組，表示A.
B.ravel() 將B 展平為一維數組，確保其元素數與A 中的行數相同。
• 索引操作 A[...] 擷取元素A 使用 B 中的行索引和列索引。

當 B 是一維陣列或列索引清單時，此方法特別有用。透過跳過展平操作，程式碼變得更簡單：

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

範例：

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