numpy.ndarray.take

метод

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

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

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

См. также

numpy.take

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