Pandas df中每个用户的两个事件行之间的时间差

import pandas as pd

# 假设df是一个包含用户、事件和时间的DataFrame
df = pd.DataFrame({
    '用户': ['A', 'A', 'B', 'B', 'B', 'C'],
    '事件': ['事件1', '事件2', '事件1', '事件2', '事件3', '事件1'],
    '时间': ['2022-01-01 10:00:00', '2022-01-01 11:00:00', '2022-01-01 12:00:00', '2022-01-01 13:00:00', '2022-01-01 14:00:00', '2022-01-01 15:00:00']
})

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

# 按照用户进行分组，并计算时间差
df['时间差'] = df.groupby('用户')['时间'].diff()

print(df)

  用户   事件                  时间      时间差
0  A  事件1 2022-01-01 10:00:00      NaT
1  A  事件2 2022-01-01 11:00:00 01:00:00
2  B  事件1 2022-01-01 12:00:00      NaT
3  B  事件2 2022-01-01 13:00:00 01:00:00
4  B  事件3 2022-01-01 14:00:00 01:00:00
5  C  事件1 2022-01-01 15:00:00      NaT