根据索引从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的方式获得输出output=[2,7,10]？

- Dzung Nguyen

Answer 1

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

或者

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

- Fred Foo

---

可能也值得一提的是 input[index == 1]。 - Bi Rico