Предположим, что у меня есть матрица A с некоторыми произвольными значениями:
array([[ 2, 4, 5, 3],
[ 1, 6, 8, 9],
[ 8, 7, 0, 2]])
И матрица B, которая содержит индексы элементов в A:
array([[0, 0, 1, 2],
[0, 3, 2, 1],
[3, 2, 1, 0]])
Как выбрать значения из A, на которые указывает B, то есть:
A[B] = [[2, 2, 4, 5],
[1, 9, 8, 6],
[2, 0, 7, 8]]