numpy.delete对面

Question

我有一个看起来像[3,65,7,83,2,4]的numpy数组，我希望保留索引[1,3,5]。哪个会给我[65, 83, 4]。 Numpy有办法做到这一点吗？

这基本上与numpy.delete函数相反。

Answer 1

使用花式索引：

>>> a = np.array([3,65,7,83,2,4])
>>> a[[1, 3, 5]]
array([65, 83,  4])