numpy.ndarray.choose

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

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

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

См. также