numpy.ndarray.choose

метод

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

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

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

См. также

numpy.choose

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