python データフレームからの抽出方法

Question

写真のようなDataFrameがある場合に、
以下の①～③のようなデータを抽出する方法を教えていただきたいです。
初歩的な内容かとは思いますが、何卒よろしくお願いします。

①最も古いtimestampの値
└写真のDataFrameのオレンジ色箇所に該当
②DataFrame内の最新のtimestampがindexになっているdata
└写真のDataFrameの緑色箇所に該当
③flag列において最新でflagが1になっているtimestamp
（ただし探索範囲は最新の時間から60秒前まで）
└写真のDataFrameの水色箇所に該当

※1：写真のDataFrameでいうと、最も古い＝0:00:00
※2：写真のDataFrameでいうと、最新　　＝0:02:10
※3：timestampがindexに指定されている
｜
｜

↓DataFrame

Answer 1

実際のコードで説明しますと、こんな感じになります。

import pandas as pd

df = pd.DataFrame({'timestamp':['0:00:00', '0:00:10', '0:00:20',
                                '0:00:30', '0:00:40', '0:00:50',
                                '0:01:00', '0:01:10', '0:01:20',
                                '0:01:30', '0:01:40', '0:01:50',
                                '0:02:00', '0:02:10'],
                   'data':[37,98,75,19,24,70,83,14,44,66,61,87,36,50],
                   'flag':[0,0,0,0,0,0,0,0,0,1,0,0,0,0]})

# timestamp列はtimedelta型(またはdatetime型)にしておく
df['timestamp'] = pd.to_timedelta(df['timestamp'])
#   timestamp  data  flag
#0   00:00:00    37     0
#1   00:00:10    98     0
#2   00:00:20    75     0
#3   00:00:30    19     0
#4   00:00:40    24     0
#5   00:00:50    70     0
#6   00:01:00    83     0
#7   00:01:10    14     0
#8   00:01:20    44     0
#9   00:01:30    66     1
#10  00:01:40    61     0
#11  00:01:50    87     0
#12  00:02:00    36     0
#13  00:02:10    50     0

### (1) 最も古いtimestampの値 ###
# timestamp列の最小値を取得
print(df['timestamp'].min())
# 0 days 00:00:00


### (2) DataFrame内の最新のtimestampがindexになっているdata ###

# timestamp列が最大値を取るindexを取得
target_index = df['timestamp'].idxmax()
# target_index行の data列を取得(.at()か .loc()を使う)
print(df.at[target_index, 'data'])
# 50

# (補足)通常は上記を纒めて以下のように書くことが多い
print(df.loc[df['timestamp'].idxmax(), 'data'])
# 50


### (3) flag列において最新でflagが1になっているtimestamp ###
###    （ただし探索範囲は最新の時間から60秒前まで）        ###

# 最新のtimestampを求める
current = df['timestamp'].max()
# 最新の時間から60秒前までの列かどうかのBoolianリストを求める
filter_list1 = (df['timestamp'] >= (current - pd.Timedelta(60, unit='s')))
# 更にflagが1の列かどうかのBoolianリストを求める
filter_list2 = (df['flag']==1)
# 上記２つの条件のANDで新しいfilter_listを作成
filter_list = (filter_list1 & filter_list2)
#上記の条件に合致した行の timestamp列を取得(.loc()を使う)
print(df.loc[filter_list, 'timestamp'])
#9   00:01:30
#Name: timestamp, dtype: timedelta64[ns]

# (補足)上記を纒めると以下のようにかける
print(df.loc[(df['timestamp'] >= (df['timestamp'].max()-pd.Timedelta(60, unit='s'))) & (df['flag']==1), 'timestamp'])
#9   00:01:30
#Name: timestamp, dtype: timedelta64[ns]

Comments

[goma-goma]

ありがとうございます！助かりました！