my array subset method could be improved?

Jim O'D jbo at cannedham.ee.ed.ac.uk
Fri Oct 14 08:46:58 EDT 2005


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



More information about the Python-list mailing list