>>> 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. -- http://mail.python.org/mailman/listinfo/python-list