использовать в состоянии numpy.select

Question

Я пытаюсь использовать "in" в numpy.select.

x = np.arange(10)
condlist = [x in [2,3,4], x>5]
choicelist = [x, x**2]
np.select(condlist, choicelist)

Есть ли способ заставить его работать?

Answer 1

Вы должны использовать isin вместо:

x = np.arange(10)
condlist = [np.isin(x, [2,3,4]), x>5]
choicelist = [x, x**2]
np.select(condlist, choicelist)

Выход:

array([ 0,  0,  2,  3,  4,  0, 36, 49, 64, 81])