numpy.delete对面

Question

我有一个numpy数组看起来像:[3,65,7,83,2,4]我想保留索引[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])