>>> from numpy import * >>> from numpy.random import * >>> N = 1000000 >>> input = sign(randn(N)) >>> a = arange(N) >>> output = zeros(N) >>> output[input < 0] = 10 * a[input < 0] >>> output[input > 0] = 20 * a[input > 0] >>> Worked fine for me. S.M.