numpy.ndarray.choose

метод

ndarray.choose(choices, out=None, mode='raise')

Использует массив индексов для создания нового массива из набора вариантов.

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

См. также

numpy.choose

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