`numpy.ndarray`有没有` `itemgetter`‘等价物？

Question

当我使用numpy.ndarray调用itemgetter时，会得到一个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但返回ndarray的numpy函数。

Answer 1

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

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

Answer 2

可能不会回答这个问题，但我会这样做

ar[[1,3]]

获取包含所需元素的numpy.ndarray对象