当我用 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 函数。
请您参考如下方法:
使用 numpy 数组，您可以通过使用列表索引数组来直接访问多个索引:
>>> x
array([1, 2, 3, 4, 5])
>>> x[[1, 3]]
array([2, 4])
