2017-08-24 48 views
-1

如果我有:如何建立索引設置爲引用數組

x = np.asarray([[1,2],[3,4],[5,6]]) 

而且我想創建:

y = np.asarray([1,4,5]) 

爲了做到這一點,我建立了一個數組如下:

inds = np.asarray([[0,0],[1,1],[2,0]]) 

,我又把它傳給x如下:

y = x[inds] 


這不會產生由行中inds索引的元素。我如何以這種方式實現這一功能,或者與此類似的時尚?

+1

'×〔元組(inds.T)]'? –

回答

1

這是什麼advanced indexing;提取的行索引和列索引爲兩個單獨的陣列,並用它們到子集的陣列:

x[inds[:,0], inds[:,1]] 
# array([1, 4, 5])