Re: [Numpy-discussion] FW: numpy array matching

[Chris B. <cb...@jp...>]

Chris Barker wrote:
> >>> t = array(['abcd','aacd','ddfa','qqcd'])
> 
> >>> t2 = t.astype(UnsignedInt8)
> 
> >>> nonzero(equal(t2,ord('c'))[:,2])
> 
> array([0, 1, 3])

OOPS. For a little better performance:

>>> nonzero(equal(t2[:,2],ord('c')))

array([0, 1, 3])


-- 
Christopher Barker,
Ph.D.                                                           
cb...@jp...                      ---           ---           ---
http://www.jps.net/cbarker          -----@@       -----@@       -----@@
                                   ------@@@     ------@@@     ------@@@
Water Resources Engineering       ------   @    ------   @   ------   @
Coastal and Fluvial Hydrodynamics -------      ---------     --------    
------------------------------------------------------------------------
------------------------------------------------------------------------