[Numpy-discussion] Efficient array equivalent to cmp(x,y)

Kim Hansen slaunger at gmail.com
Thu Sep 17 16:33:24 EDT 2009


>
>
>
> In [1]: a = np.array([0, 2, 4, 1, 3, 0, 3, 4, 0, 1])
>
> In [2]: lim = 2
>
> In [3]: np.sign(a - lim)
> Out[3]: array([-1,  0,  1, -1,  1, -1,  1,  1, -1, -1])
>

Dooh. Facepalm. I should have thought of that myself! Only one intermediate
array needs to be created then. Thank you. That was what I was looking for.

Cheers,
Kim
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20090917/5d443c80/attachment.html>


More information about the NumPy-Discussion mailing list