如何使用 Pandas 获得滚动值计数（频率）？ （计算效率高，无循环）

Question

我有一个值列表，我想得到它们的滚动频率，所以是这样的：

df = pd.DataFrame({
    'val': [1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3]
})

result = df.val.rolling(3).freq()

result == pd.Series([1, 2, 3, 3, 3, 1, 2, 3, 3, 3, 1, 2, 3, 3, 3])

当然，我可以使用循环来执行此操作，但是对于大量数据，计算成本可能很高，因此我更愿意使用内置的或矢量化的东西等。但不幸的是，从我的搜索中，没有这似乎是一个解决方案。

提前致谢！

Answer 1

窗口大小为n 的滚动函数结果的第一个n-1 元素必须为每个定义NaN。

result = df.val.rolling(3).apply(lambda x: np.count_nonzero(x==x.iloc[-1])).astype('Int64')

结果：

0     <NA>
1     <NA>
2        3
3        3
4        3
5        1
6        2
7        3
8        3
9        3
10       1
11       2
12       3
13       3
14       3