Выберите один элемент в каждой строке массива Numpy по индексам столбцов
Есть ли лучший способ получить "output_array" из "input_array" и "select_id"?
Можем ли мы избавиться отrange( input_array.shape[0] ) ?
>>> input_array = numpy.array( [ [3,14], [12, 5], [75, 50] ] )
>>> select_id = [0, 1, 1]
>>> print input_array
[[ 3 14]
[12 5]
[75 50]]
>>> output_array = input_array[ range( input_array.shape[0] ), select_id ]
>>> print output_array
[ 3 5 50]