特定の行を中心に前後t分間を別ファイルに抽出

Question

前提・実現したいこと

下記のようなcsvファイルにおいて，特定の行(ここでは，google.com)を発見したら，そこを中心に前後t分間を抽出して別ファイルにquery列のものだけを出力する．という動作をpythonで実現したいと考えています．日付に関しては，unixtimeに変換しています．

　date_time 　info 　　client 　　IP 　　　　 query
0 1604223469 info: client 10.3.10.202 yahoo.com
1 1604223469 info: client 10.3.10.202 hogehoge.com
2 1604223470 info: client 10.3.10.202 fugafuga.com
3 1604223471 info: client 10.3.10.202 japan.com
4 1604223471 info: client 10.3.10.202 google.com
5 1604223471 info: client 10.3.10.202 fire.com
6 1604223472 info: client 10.3.10.202 super.com
7 1604223472 info: client 10.3.10.202 domain.com
8 1604223472 info: client 10.3.10.202 spanixh.com
9 1604223472 info: client 10.3.10.202 amazon.com

ただ，条件指定や時間での切り取り等，動作処理をどのように行うのか分からずに苦悩しています．
アドバイスをいただきたく質問させてもらいました．

Answer 1

pandasを使えば以下のような感じで抽出できます。

import pandas as pd
from io import StringIO
from datetime import datetime, timedelta

# テストデータ
s = """date_time info client IP query
1604223469 info: client 10.3.10.202 yahoo.com
1604223469 info: client 10.3.10.202 hogehoge.com
1604223470 info: client 10.3.10.202 fugafuga.com
1604223471 info: client 10.3.10.202 japan.com
1604223471 info: client 10.3.10.202 google.com
1604223471 info: client 10.3.10.202 fire.com
1604223472 info: client 10.3.10.202 super.com
1604223472 info: client 10.3.10.202 domain.com
1604223472 info: client 10.3.10.202 spanixh.com
1604223472 info: client 10.3.10.202 amazon.com"""
df = pd.read_csv(StringIO(s), delimiter='\s+')
df['date_time'] = pd.to_datetime(df['date_time'], unit='s')

base = pd.to_datetime(df[df['query'] == 'google.com']['date_time'].values[0])
delta = timedelta(seconds=1) # 前後1秒
st, ed = base - delta, base + delta
mask = (df['date_time'] >= st) & (df['date_time'] <= ed)
ret = df[mask]
print(ret)
"""
            date_time   info  client           IP         query
2 2020-11-01 09:37:50  info:  client  10.3.10.202  fugafuga.com
3 2020-11-01 09:37:51  info:  client  10.3.10.202     japan.com
4 2020-11-01 09:37:51  info:  client  10.3.10.202    google.com
5 2020-11-01 09:37:51  info:  client  10.3.10.202      fire.com
6 2020-11-01 09:37:52  info:  client  10.3.10.202     super.com
7 2020-11-01 09:37:52  info:  client  10.3.10.202    domain.com
8 2020-11-01 09:37:52  info:  client  10.3.10.202   spanixh.com
9 2020-11-01 09:37:52  info:  client  10.3.10.202    amazon.com
"""

Comments

[taraco_040103]

丁寧な回答ありがとうございます．
また，回答に対する質問なのですが
date_time 　info 　　client 　　IP 　　　　 query
0 1604223469 info: client 10.3.10.202 yahoo.com
.
.
.
といったデータがsample.csvというファイルの中にある場合は，変数sにsample.csvの中身を渡すという形でプログラミングを書いていくと良いでしょうか？

[taraco_040103]

重ねての質問申し訳ないですが，回答いただけると幸いです．

[8524ba23]

df = pd.read_csv('sample.csv', ～)という感じでよいです。詳しくは以下を参照ください。
https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_csv.html

[taraco_040103]

早急なご対応ありがとうございます．頂いた回答共々，参考にさせていただきます．