numpy.select

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

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

Параметры:

condlist : list of bool ndarrays Список условий, определяющих, из какого массива в choicelist берутся элементы вывода. Когда удовлетворяется несколько условий, используется первое встретившееся в condlist. choicelist : list of ndarrays Список массивов, из которых берутся элементы вывода. Длина должна быть такой же, как у condlist. default : scalar, optional Элемент, вставляемый в output, когда все условия равны False.

Возвращает:

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

См. также

where

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

take, choose, compress, diag, diagonal

Примеры

>>> x = np.arange(10)
>>> condlist = [x<3, x>5]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist)
array([ 0,  1,  2,  0,  0,  0, 36, 49, 64, 81])