矩阵索引，提取每个不同列

Question

底部有小数组的示例。 我有一个维数为MxN的numpy数组数据（N>；100）和尺寸为Mx100的较小阵列

数组A包含数据数组的列指示符

我想提取由A中的索引给出的所有数据列。 问题是每一行的索引都不同

我试过了

 data[:,A]

但这会返回一个MX100数组。 我只想要一个Mx100阵列。 我可以做一个for循环，但这需要很长时间。 我怎样才能做到这一点

示例：

data=np.array([[1,2,3],[4,5,6],[7,8,9]])
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
A=np.array([[0,1],[1,2],[0,2]])

我想提取： 第一行：索引0和1。第二行：索引1和2。第三行：索引0和2

期望结果：

array([[1,2],
[5,6],
[7,9]])

Answer 1

使用broadcasting

>>> data[np.arange(A.shape[0])[:,None], A]

array([[1, 2],
       [5, 6],
       [7, 9]])