numpy.select

numpy.select(condlist, choicelist, default=0)[source]

Возвращает массив, элементы которого выбираются из массивов в списке choicelist в зависимости от условий.

Параметры:

condlistсписок bool ndarrays

Список условий, определяющих, из какого массива в списке choicelist берутся элементы для выходного массива. Если несколько условий выполняются, используется первое из них.

choicelistсписок ndarrays

Список массивов, из которых берутся элементы выходного массива. Длина списка должна быть такой же, как у condlist.

defaultскаляр, необязательно

Элемент, вставляемый, когда все условия ложны.

Возвращаемое значение:

outputndarray

Элемент в позиции m — это m-й элемент массива из choicelist, где m-й элемент соответствующего массива в condlist равен True.

См. также

numpy.where

Возвращает элементы из одного из двух массивов в зависимости от условия.

numpy.take, numpy.choose, numpy.compress, numpy.diag, numpy.diagonal

Примеры

Начиная с массива целых чисел от 0 до 5 (включительно), элементы меньше 3 заменяются на противоположные, элементы больше 4 возводятся в квадрат, а элементы, которые не удовлетворяют ни одному из этих условий (ровно 3), заменяются значением по умолчанию, равным 0.

>>> x = np.arange(6)
>>> condlist = [x<3, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 42)
array([ 0,  1,  2, 42, 16, 25])

Когда несколько условий выполняются, используется первое из них.

>>> condlist = [x<=4, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 55)
array([ 0,  1,  2,  3,  4, 25])