numpy.select

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

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

Параметры

condlistсписок bool ndarrays

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

choicelistсписок ndarrays

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

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

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

Возвращает

outputndarray

Элемент в позиции 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, ..., 49, 64, 81])