обтяжные элементы из 2D-массива на основе индекса массива

Question

У меня есть массив, как:

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

У меня также есть массив индексов, как:

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

Элементы Ind представляют номер строки A для каждого столбца A.

т.е.

Я хочу, чтобы вытащить ind[0] = 1 элемент из столбца 0 из A Я хочу, чтобы вытащить ind[4] = 2 элемент из колонки 4 A

Желаемой выходе:

5, 11, 2, 13, 14 

Answer 1

Использование Numpy's fancy-indexing -

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