獲取與numpy的TAKE功能

Question

我有一個一個維數組（n）的多個維度稱爲邊緣，並希望通過從頂點數組的索引中插入值（N，3）

vertices = [[ 1.25, 4.321, -4], [2, -5, 3.32], [23.3, 43, 12], [32, 4, -23]] 

edges = [1, 3, 2, 0] 


result = [[2, -5, 3.32], [32, 4, -23], [23.3, 43, 12], [ 1.25, 4.321, -4]] 

我嘗試了np.take（頂點，邊），但它不適用於多維數組。

Answer 1

take與軸參數工作

In [313]: vertices=np.array(vertices) 
In [314]: edges=[1,3,2,0] 
In [315]: np.take(vertices, edges,0) 
Out[315]: 
array([[ 2. , -5. , 3.32 ], 
     [ 32. , 4. , -23. ], 
     [ 23.3 , 43. , 12. ], 
     [ 1.25 , 4.321, -4. ]]) 
In [316]: vertices[edges,:] 
Out[316]: 
array([[ 2. , -5. , 3.32 ], 
     [ 32. , 4. , -23. ], 
     [ 23.3 , 43. , 12. ], 
     [ 1.25 , 4.321, -4. ]]) 

Answer 2

您可以簡單地使用索引這裏：

vertices[edges] 
#  ^ ^indexing

如果指數與列表，則numpy的將洗牌原矩陣由edges規定使得最高層面這裏遵循的指標。

喜歡：

>>> vertices = np.array([[ 1.25, 4.321, -4], [2, -5, 3.32], [23.3, 43, 12], [32, 4, -23]]) 
>>> edges = [1, 3, 2, 0] 
>>> vertices[edges] 
array([[ 2. , -5. , 3.32 ], 
     [ 32. , 4. , -23. ], 
     [ 23.3 , 43. , 12. ], 
     [ 1.25 , 4.321, -4. ]]) 
>>> vertices[edges].base is None 
True 

事實上，base是None意味着這並不生成視圖，它使基質的副本（與過濾/重新排序行）。您稍後對元素vertices所作的更改不會更改vertices[edges]結果的元素（假設您在更改vertices之前進行復制）。