array logical operations in python?

zero79 · 05-12-2005, 03:00 PM

Hi all,

Are there any python libraries that implement functionality similar to logical array operations in MATLAB? For example in matlab

Code:

x = 1:10
x(x>5) = 0

will output

Code:

x =

  1  2  3  4  5  0  0  0  0  0

where x>5 gives the logical array where x is greater than 5

Code:

x>5 =
  0 0 0 0 0 1 1 1 1 1

I know I can accomplish the same goal with for/if/else, but array operations should be significantly faster. Thanks for any thoughts.

zero79 · 05-12-2005, 03:14 PM

Ok, I think I found what I am looking for

Code:

#!/usr/bin/python

from Numeric import *
x = array((1,2,3,4,5,6,7,8,9,10))
print where(greater(x,5),x,0)

ag2uki · 05-13-2005, 03:56 AM

It could be like this....

x = range(1,10)
x = [j for j in (x[:5]+[0 for k in range(4)])]

regards
Agus Dwi Basuki