numpy.select#

numpy.select(condlist, choicelist, по умолчанию=0)[источник]#

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

Параметры:

condlistсписок bool ndarrays: Список условий, которые определяют, из какого массива в choicelist выбираются выходные элементы. Когда удовлетворены несколько условий, выбирается первое встреченное в condlist используется.
choicelistсписок ndarrays: Список массивов, из которых берутся выходные элементы. Должен иметь ту же длину, что и condlist.
по умолчаниюскаляр, опционально: Вставленный элемент в вывод когда все условия оцениваются как False.

Возвращает:

выводndarray: Выходное значение на позиции m является m-м элементом массива в choicelist где m-й элемент соответствующего массива в condlist равно True.

Смотрите также

where: Возвращает элементы из одного из двух массивов в зависимости от условия.
take, choose, compress, diag, diagonal

Примеры

>>> import numpy as np

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

>>> 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 используется.

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