python - NumpyRecArrayから行を選択します

Question

Numpy rec配列があり、そこからSQLに似た簡単なクエリを実行したいと思いますSELECT * where array['phase'] == "P"。クエリ条件を満たす元の配列の行に対応する各行を含むレコード配列を出力として取得したいと思います。何か案は？私は以前にこれを行ったことがあると確信していますが、機能を思い出せません。

ありがとう

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')])

Answer 1

試す：

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

array['phase']=='P'ブール値のnumpy配列を返します。idxがブール配列の場合、array[idx]はこれらの行で構成される配列を返します。ここで、idxはTrueです。

Answer 2

このコードを試してください：

array[array.phase=='P']