numpy.matrix.take

метод

matrix.take(indices, axis=None, out=None, mode='raise')

Возвращает массив, образованный из элементов a по заданным индексам.

Обратитесь к numpy.take для полной документации.

См. также

numpy.take

эквивалентная функция