從基於索引的numpy數組中獲取子數組

Question

我有一個numpy數組向量，我想得到一個基於索引的子集：

import numpy as np
input=np.array([1,2,3,4,5,6,7,8,9,10])
index=np.array([0,1,0,0,0,0,1,0,0,1])

什么是pythonic方法來輸出輸出= [2,7,10]？

Answer 1

output = input[index.astype(np.bool)]

要么

output = input[np.where(index)[0]]