基于索引数组从2D数组中提取元素

Question

我有一个数组：

import numpy as np
A = np.arange(15).reshape(3, 5)

我还有一个索引数组：

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

ind的元素表示A的每一列的A行号。

即

我想从A的第0列拉出ind[0] = 1元素 我想从A

的第4列中提取ind[4] = 2元素

所需的输出是：

5, 11, 2, 13, 14

Answer 1

使用Numpy's fancy-indexing -

A[ind,np.arange(ind.size)]