numpyの一列目に特定の値をもつ行を抽出したい。

Question

Pyhtonのnumpyの特定の配列を抽出、または削除したいです。

発生している問題・エラーメッセージ

array([[   2,    1,    0, ...,   -1,   -1,   -1],
       [   2,    2,    2, ...,   -1,   -1,   -1],
       [   2,    3,    9, ...,   -1,   -1,   -1],
       ...,
       [1561,   99,    0, ...,   -1,   -1,   -1],
       [1561,  107,    2, ...,   -1,   -1,   -1],
       [1561,  110,    0, ...,   -1,   -1,   -1]])

物体検出のライブラリを用いて、映像から物体検出を行い、エクセルのデータにし、それを読み込み、配列にしました。
一列目の数字がフレームの番号となっております。
60fpsの映像であり、0.5秒、つまり30フレームごとに配列を抜き出したい(つまり一列目の数字が2,32,62,,,,となる行が欲しい)のですが、どうしたら良いのかわかりません。

該当のソースコード

import numpy as np
import random
import pandas as pd
import openpyxl
import xlrd

df = pd.read_excel("mati_kekka.xlsx", sheet_name='Sheet1')

nmp=df.to_numpy()
syudan = nmp

syudan

試したこと

いろいろ調べてみたのですが、よくわからず、質問させていただきました。
よろしくお願いいたします。

Answer 1

先頭を30で割ったあまりが2の行を取り出したいと考えると、次の式でできないでしょうか。

#
nmp[nmp[:, 0] % 30 == 2]

Answer 2

単純な方法ですが、以下のようなループで目的の番号だけを持つ配列を抽出できます。

import numpy as np

# １行以上あり、0列目は整数で、その昇順で並んでいるものとする
src = np.array([[2,21],[2,22],[3,31],[32,321]])

# 先頭の値から末尾の行の値まで30刻み
for no in range(src[0][0],src[src.shape[0]-1][0]+1,30):
    sub = src[src[:,0] == no]
    print('-----\n',sub)

"""
-----
 [[ 2 21]
 [ 2 22]]
-----
 [[ 32 321]]
"""

Comments

[tontonchin]

ありがとうございます。一旦試してみます。

Answer 3

みなさん、本当に回答ありがとうございます。
しかし、あまりプログラミングがわからず、(初心者なので他の人のプログラムが読めない)なので、みなさんの提示してくださった方法から自己流で次のように書いて解決しました。ご回答いただいた方本当にありがとうございました。

syudanは元の配列

arr = np.empty((0,7), int)
for i in range(10523):
    f = (syudan[i,0] - 2)/30
    if f.is_integer():
        print(syudan[i,0])
        brr =np.append(brr,np.array([[syudan[i,0],syudan[i,1],syudan[i,2],syudan[i,3],syudan[i,4],syudan[i,5],syudan[i,6]]]), axis=0)