numpy.ndarray.take

метод

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

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

См. numpy.take для полной документации.

См. также

numpy.take

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