如何查找分组中两个事件之间的时间差

import pandas as pd

# 示例数据
data = {
    'group': ['A', 'A', 'B', 'B'],
    'event': ['start', 'end', 'start', 'end'],
    'timestamp': ['2023-04-01 10:00:00', '2023-04-01 10:05:00', '2023-04-01 11:00:00', '2023-04-01 11:10:00']
}

df = pd.DataFrame(data)
df['timestamp'] = pd.to_datetime(df['timestamp'])

# 计算时间差
time_diffs = df.groupby('group').apply(lambda x: (x[x['event'] == 'end']['timestamp'].iloc[0] - x[x['event'] == 'start']['timestamp'].iloc[0]).total_seconds())

print(time_diffs)

df['timestamp'] = pd.to_datetime(df['timestamp'], errors='coerce')

df.sort_values(by=['group', 'timestamp'], inplace=True)

df = df.dropna(subset=['event', 'timestamp'])