Groupby列，按时间戳排序，并计算Pandas Dataframe中时间戳之间的差异？

import pandas as pd

# 创建示例数据
data = {'时间戳': ['2022-01-01 10:00:00', '2022-01-01 10:05:00', '2022-01-01 10:10:00', '2022-01-01 10:15:00'],
        '数值': [1, 2, 3, 4],
        '分组列': ['A', 'A', 'B', 'B']}
df = pd.DataFrame(data)

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

# 按分组列进行分组，并按时间戳排序
df_sorted = df.groupby('分组列').apply(lambda x: x.sort_values('时间戳'))

# 计算时间戳之间的差异
df_sorted['时间差'] = df_sorted['时间戳'] - df_sorted['时间戳'].shift(1)

# 打印结果
print(df_sorted)