如何收集numpy中特定索引的元素？

Question

我想收集指定轴中指定索引的元素，如下所示。

x = [[1,2,3], [4,5,6]] 
index = [[2,1], [0, 1]] 
x[:, index] = [[3, 2], [4, 5]] 

这实质上是在pytorch中的收集操作，但正如你所知道的，这在numpy中是不可行的。我想知道在numpy中是否有这样的“聚集”操作？

Answer 1

>>> x = np.array([[1,2,3], [4,5,6]]) 
>>> index = np.array([[2,1], [0, 1]]) 
>>> x_axis_index=np.tile(np.arange(len(x)), (index.shape[1],1)).transpose() 
>>> print x_axis_index 
[[0 0] 
[1 1]] 
>>> print x[x_axis_index,index] 
[[3 2] 
[4 5]]