Numpyで特定の列/行に条件を指定してマッチする行/列全体を抜きだしたい - Qiita

お悩み numpyで特定の列に対して条件を指定して行全体を操作したいことがある。 例えば、機械学習のモデルに画像を通すと、オブジェクトクラスの値がnumpyの中に入って返ってくるけれど、特定のクラスだけフィルタかけたい時とか。本筋としてはモデル側で欲しいクラスだけを返すようにしろって話なんだけど、それができない時の話。 下の出力例だと6番目の値がオブジェクトクラスなので、例えば車(2)だけの結果を集めたいとか、人間(0)だけを集めたい時に、6列目の値で検索して該当する行だけ集めたい。 [[5.39797363e+02 2.71583344e+02 5.93862732e+02 3.26177856e+02 8.78980994e-01 2.00000000e+00] [7.61696655e+02 2.84762238e+02 7.84034058e+02 3.39492310e+02 8