Select Rows from Numpy Rec Array

Question

I have a Numpy rec array from which I would like to do some quick queries similar to SQL: SELECT * where array['phase'] == "P". I would like to get a Record Array as output with each row corresponding to a row from the original array that met the query criteria. Any ideas? I am pretty sure I have done this before, but just cannot remember the function.

Thanks

rec.array([ (5295499, 2.8123965, 127.20443, 0.0, 1237680436.06, 908, -19.942589, 134.33951, 0.3888, 'P', 0.19513991),
       (5295499, 2.8123965, 127.20443, 0.0, 1237680436.06, 1387, -18.102, 125.639, 0.11, 'P', 1.2515257),
       (5447254, 39.025873, 143.31065, 0.0, 1245455521.85, 1512, 33.121667, 130.87833, 0.573, 'LR', 45.099504)], 
      dtype=[('eventid', '<i4'), ('eventlat', '<f8'), ('eventlon', '<f8'), ('eventdepth', '<f8'), ('eventtime', '<f8'), ('stationid', '<i4'), ('stationlat', '<f8'), ('stationlon', '<f8'), ('stationelv', '<f8'), ('phase', '|S7'), ('timeresidual', '<f8')])

unutbu · Accepted Answer · 2011-09-06 00:46:32Z

9

Try:

array[array['phase']=='P']

array['phase']=='P' returns a boolean numpy array. When idx is a boolean array, array[idx] returns an array composed of those rows where idx is True.

edited Sep 6, 2011 at 0:46

answered Sep 6, 2011 at 0:22

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Rishi Over a year ago

This is genius! But just to clarify for other ppl: array here does not refer to numpy.array, it refers to the name of your record array. For instance, if I had a recarray called data, then I would want to do: data[data['phase'] == 'P']]

CuberChase · Accepted Answer · 2013-07-05 02:59:35Z

0

Try this code:

array[array.phase=='P']

edited Jul 5, 2013 at 2:59

CuberChase

4,5185 gold badges36 silver badges52 bronze badges

answered Jul 5, 2013 at 2:35

Rakhesh Devadas

1

Collectives™ on Stack Overflow

Select Rows from Numpy Rec Array

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related