使用NumPy 數組作為另一個數組第二維的索引
根據數組提供的索引從2D 數組中提取特定元素第二個數組，您可以利用NumPy的整數陣列
考慮這個範例：
A = np.array([[0,1], [2,3], [4,5]]) B = np.array([[1], [0], [1]], dtype='int')
要得到以下所需的輸出：
C = np.array([[1], [2], [5]])
您可以使用以下方法：
A[np.arange(A.shape[0]),B.ravel()]
怎麼樣有效：
或者，如果B 是一維數組或列索引列表，則可以使用.ravel() 跳過展平：
A[np.arange(A.shape[0]),B]
此方法提供了一種使用從另一個數組派生的索引從2D 數組中提取元素的簡單方法。
以上是如何使用另一個數組的索引從二維數組中提取元素？的詳細內容。更多資訊請關注PHP中文網其他相關文章！
