计算pandas数据帧中的重叠时间帧，按人员分组

import pandas as pd

# 创建包含时间戳和人员信息的数据帧
data = {'timestamp': ['2022-01-01 09:00:00', '2022-01-01 09:05:00', '2022-01-01 09:10:00', '2022-01-01 09:15:00'],
        'person': ['A', 'B', 'A', 'B']}
df = pd.DataFrame(data)

# 将时间戳列转换为Datetime类型
df['timestamp'] = pd.to_datetime(df['timestamp'])

# 按人员进行分组
grouped = df.groupby('person')

# 定义函数来计算重叠时间帧
def calculate_overlap(group):
    # 向前移动时间戳列
    group['previous_timestamp'] = group['timestamp'].shift()
    
    # 合并原始数据帧和移动后的数据帧
    merged = pd.merge(group, group, left_on='timestamp', right_on='previous_timestamp')
    
    # 筛选出重叠的时间帧
    overlap_frames = merged[merged['timestamp_x'] >= merged['previous_timestamp_y']]
    
    # 返回重叠时间帧
    return overlap_frames

# 应用函数到每个分组
overlap_frames_by_person = grouped.apply(calculate_overlap)

# 打印结果
print(overlap_frames_by_person)