计算每个gorup的两个不同行之间的时间差，并在一列中使用“lag”

import pandas as pd

# 创建示例数据
data = {
    'group': ['A', 'A', 'A', 'B', 'B', 'B'],
    'timestamp': ['2023-01-01 12:00:00', '2023-01-01 12:01:00', '2023-01-01 12:02:00',
                  '2023-01-01 12:00:00', '2023-01-01 12:01:00', '2023-01-01 12:02:00']
}
df = pd.DataFrame(data)

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

# 计算 lag 时间差
df['lag'] = df.groupby('group')['timestamp'].diff().dt.total_seconds()

print(df)

  group           timestamp        lag
0     A 2023-01-01 12:00:00       NaN
1     A 2023-01-01 12:01:00  60.000000
2     A 2023-01-01 12:02:00  60.000000
3     B 2023-01-01 12:00:00       NaN
4     B 2023-01-01 12:01:00  60.000000
5     B 2023-01-01 12:02:00  60.000000