[英]Numpy: Indexing 3D matrix using 1D array
arr = np.arange(12).reshape((3, 2, 2))
indices = np.array([0, 1, 1])
expected_outcome = np.array([[0, 1], [6, 7], [10, 11]])
我正在嘗試使用包含我想要獲取的值的 y 索引的形狀數組 (3) 來索引這個形狀數組 (3,2,2)。 我試圖讓它與for in
語句一起工作,但有沒有一種優雅的方法可以用 numpy 做到這一點?
所以你想要arr[0,0,:], arr[1,1,:], arr[2,1,:]
?
怎么樣
In [179]: arr[[0,1,2], [0,1,1]]
Out[179]:
array([[ 0, 1],
[ 6, 7],
[10, 11]])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.