如何使用另一個包含索引的numpy數組索引一個numpy數組？

Question

這應該是簡單的任務，但我很慚愧地承認自己被卡住了。

我有一個numpy的陣列，稱爲X：

X.shape是(10,3)，它看起來像

[[ 0. 0. 13. ] 
[ 0. 0. 1. ] 
[ 0. 4. 16. ] 
..., 
[ 0. 0. 4. ] 
[ 0. 0. 2. ] 
[ 0. 0. 4. ]] 

我想選擇這個數組的1，第2和第3排，使用指數在這個其他numpy的陣列，稱爲idx：

idx.shape爲(3,)，它看起來像[1 2 3]

當我嘗試
new_array = X[idx]或此變化， 我收到錯誤。

如何使用另一個包含索引的numpy數組索引一個numpy數組？

提前爲這樣一個基本問題道歉。

Answer 1

我不喜歡這樣寫道：

>>> import numpy as np 
>>> x = np.arange(30).reshape((10, 3)) 
>>> x 
array([[ 0, 1, 2], 
     [ 3, 4, 5], 
     [ 6, 7, 8], 
     [ 9, 10, 11], 
     [12, 13, 14], 
     [15, 16, 17], 
     [18, 19, 20], 
     [21, 22, 23], 
     [24, 25, 26], 
     [27, 28, 29]]) 
>>> idx = np.array([1,2,3]) 
>>> x[idx, ...] 
array([[ 3, 4, 5], 
     [ 6, 7, 8], 
     [ 9, 10, 11]]) 

注意，在這種情況下，省略號可以通過一個簡單的切片取代，如果你寧願：

x[idx, :]