Функция Numeric.choose()
Эта функция использует один массив с целыми числами от 0 до n для выбора значения из одного из заданных массивов:
>>> a = Numeric.identity(4) >>> b0 = Numeric.reshape(Numeric.arrayrange(16), (4, 4)) >>> b1 = -Numeric.reshape(Numeric.arrayrange(16), (4, 4)) >>> print Numeric.choose(a, (b0, b1)) [[ 0 1 2 3] [ 4 -5 6 7] [ 8 9 -10 11] [ 12 13 14 -15]]
