python - `numpy.ndarray` 是否有任何 `itemgetter` 等价物？

Question

当我itemgetter用 a调用时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 对象