my array subset method could be improved?

[Juho Schultz]

Jim O'D wrote:
> Hi all
> 
> I have an array a=array([2,3,1]).
> 
> I want to extract an array with all the elements of a that are less than 0.
> 
> Method 1.
> new = array([i for i in a if i < 0])
> 
> Method 2.
> new = a[nonzero(a<0)]
> 
> I'm using Numeric arrays but can't seem to find a function that does this.
> 
> Am I missing a more obvious way to do it quickly?
> 
> Thanks
> 
> Jim

new = Numeric.compress(Numeric.less(a,0),a)