[Python]データフレームにおける特定行の抽出

Question

位置情報データを扱っています。
データ量は約数千万レコードです。

"千代田区"に滞在したユーザを対象に、
"千代田区滞在前と後"それぞれの全データを"ユーザ毎"に抽出したいのですが、
なかなかうまくいきません。

知見のある方、ご教示いただけないでしょうか。
宜しくお願い致します。

【df】
Index｜　 id　｜　Timestamp　｜　citycode(=市区町村コード)
1　　AA1　　2022/06/10 11:07　　001(品川区)　※適当です
2　　AA1　　2022/06/10 11:22　　002(港区)
3　　AA1　　2022/06/10 12:10　　003(千代田区)
4　　AA1　　2022/06/10 14:04　　003(千代田区)
5　　AA1　　2022/06/10 14:41　　004(新宿区)
6　　AA1　　2022/06/10 18:24　　005(渋谷区)
7　　BB4　　2022/06/11 06:06　　006(目黒区)
8　　BB4　　2022/06/11 07:58　　006(目黒区)
9　　BB4　　2022/06/11 08:37　　003(千代田区)
10　　BB4　　2022/06/11 17:50　　003(千代田区)
11　　BB4　　2022/06/11 20:25　　006(目黒区)
12　　BB4　　2022/06/11 22:33　　006(目黒区)

　　　　　　　　 ↓
【千代田区滞在前のデータ抽出】
1　　AA1　　2022/06/10 11:07　　001(品川区)　※適当です
2　　AA1　　2022/06/10 11:22　　002(港区)
3　　BB4　　2022/06/11 06:06　　006(目黒区)
4　　BB4　　2022/06/11 07:58　　006(目黒区)

【千代田区滞在後のデータ抽出】
1　　AA1　　2022/06/10 14:41　　004(新宿区)
2　　AA1　　2022/06/10 18:24　　005(渋谷区)
3　　BB4　　2022/06/11 20:25　　006(目黒区)
4　　BB4　　2022/06/11 22:33　　006(目黒区)

Answer 1

import pandas as pd
import io

pd.set_option('display.unicode.east_asian_width', True)

csv_data = '''
id,Timestamp,citycode
AA1,2022/06/10 11:07,001(品川区)
AA1,2022/06/10 11:22,002(港区)
AA1,2022/06/10 12:10,003(千代田区)
AA1,2022/06/10 14:04,003(千代田区)
AA1,2022/06/10 14:41,004(新宿区)
AA1,2022/06/10 18:24,005(渋谷区)
BB4,2022/06/11 06:06,006(目黒区)
BB4,2022/06/11 07:58,006(目黒区)
BB4,2022/06/11 08:37,003(千代田区)
BB4,2022/06/11 17:50,003(千代田区)
BB4,2022/06/11 20:25,006(目黒区)
BB4,2022/06/11 22:33,006(目黒区)
'''
df = pd.read_csv(io.StringIO(csv_data), parse_dates=['Timestamp'])

#
dfx = df.groupby('id').apply(lambda x:
          x[x['Timestamp'] < x.loc[x['citycode'].eq('003(千代田区)'), 'Timestamp'].min()])\
        .reset_index(drop=True)

dfy = df.groupby('id').apply(lambda x:
          x[x['Timestamp'] > x.loc[x['citycode'].eq('003(千代田区)'), 'Timestamp'].max()])\
        .reset_index(drop=True)

print(dfx)
print(dfy)

#
    id           Timestamp     citycode
0  AA1 2022-06-10 11:07:00  001(品川区)
1  AA1 2022-06-10 11:22:00    002(港区)
2  BB4 2022-06-11 06:06:00  006(目黒区)
3  BB4 2022-06-11 07:58:00  006(目黒区)

    id           Timestamp     citycode
0  AA1 2022-06-10 14:41:00  004(新宿区)
1  AA1 2022-06-10 18:24:00  005(渋谷区)
2  BB4 2022-06-11 20:25:00  006(目黒区)
3  BB4 2022-06-11 22:33:00  006(目黒区)

Comments

[sanfuji]

回答ありがとうございます。助かります。
記載いただいたコードの実行にて、イメージ通りのデータ抽出を確認しました。