Python Numpy 根據索引值列表提取數組元素

Question

我有以下 numpy 數組：

X = np.array([1,1,0,1,1,0,1,0,0,1])

我有一個列表形式的索引值列表：

idx_list = [0,4,5,8,9]

我想要一個 numpy 數組，其中原始數組的元素對應於這些索引值（以相同的順序）。

所以這就是我想要的：

X_sub = np.array([1,1,0,0,1])

我試過這樣做：

X_sub = X[idx_list]

但出現錯誤：

KeyError: "Passing list-likes to .loc or [] with any missing labels is no longer supported.

我究竟做錯了什么？

Answer 1

多謝你們。 愚蠢的我。 它與 pandas 相關。 答案是你們中的一個人建議使用 loc/iloc。