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

Question

1

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

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

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

itthrill 31 авг. 2018, в 20:26

Теги:

numpy

1 ответ

PMende · Accepted Answer · 2018-08-31T16-02-00.000Z

Вместо этого вы должны использовать 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])