对于`numpy.ndarray`，有没有`itemgetter`等效？

Question

当我使用itemgetter致电numpy.ndarray时，我会收到tuple。

In [1]: import numpy as np                                                    
In [2]: import operator as op                                                 
In [3]: ar = np.array([1,2,3,4,5])                                            
In [4]: op.itemgetter(1,3)(ar)                                                
Out[4]: (2, 4) 

我想知道是否有任何类似于itemgetter的numpy函数，而是返回ndarray。

Answer 1

使用numpy数组，您可以通过使用列表索引到数组中来直接访问多个索引：

>>> x
array([1, 2, 3, 4, 5])
>>> x[[1, 3]]
array([2, 4])

Answer 2

可能没有回答这个问题，但我会做类似

的事情

ar[[1,3]]

获取带有必需元素的numpy.ndarray对象