事件序列的Pandas动态滚动计数

import pandas as pd

# 创建示例数据
data = pd.DataFrame({
    '时间': pd.date_range('2022-01-01', periods=10, freq='1T'),
    '事件': ['A', 'A', 'B', 'A', 'C', 'B', 'C', 'D', 'A', 'B']
})

# 按时间排序
data = data.sort_values('时间')

# 计算滚动窗口内事件的计数
data['动态滚动计数'] = data['事件'].rolling('3T').count()

print(data)

                   时间  事件  动态滚动计数
0 2022-01-01 00:00:00  A      1.0
1 2022-01-01 00:01:00  A      2.0
2 2022-01-01 00:02:00  B      3.0
3 2022-01-01 00:03:00  A      3.0
4 2022-01-01 00:04:00  C      3.0
5 2022-01-01 00:05:00  B      3.0
6 2022-01-01 00:06:00  C      3.0
7 2022-01-01 00:07:00  D      3.0
8 2022-01-01 00:08:00  A      3.0
9 2022-01-01 00:09:00  B      3.0