索引多維 numpy 數組中的多個元素

問題內容

我想使用另一個索引陣列來提取給定多維 numpy 陣列的元素。但是它的行為並不符合我的預期。下面是一個簡單的例子：

import numpy as np

a = np.random.random((3, 3, 3))
idx = np.asarray([[0, 0, 0], [0, 1, 2]])

b = a[idx]
print(b.shape)  # expect (2, ), got (2, 3, 3, 3)

為什麼會這樣呢？我應該如何修改程式碼以使 b 只包含兩個元素：a[0, 0, 0] 和 a[0, 1, 2]？

正確答案

您正在尋找 numpy 進階索引

#https://www.php.cn/link/2d661a763280f48803f3c9ba8ba0e00b

在您的情況下，您需要在每個軸上使用 idx：

a[idx[:,0], idx[:, 1], idx[:, 2]].shape == (2,)  # True

以上是索引多維 numpy 數組中的多個元素的詳細內容。更多資訊請關注PHP中文網其他相關文章！