2012-12-26 60 views
1

當我撥打itemgetternumpy.ndarray,我得到一個tuple是否有任何`itemgetter`等同於`numpy.ndarray`?

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) 

我不知道是否有任何numpy的功能就像itemgetter但返回ndarray代替。

回答

3

隨着numpy的陣列可以直接通過索引訪問多個索引到數組列表:

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

可能不會回答這個問題,但我想這樣做

ar[[1,3]] 

找回具有所需元素的numpy.ndarray對象

相關問題