Pandas 滚动应用函数到整个窗口数据框

Question

我想将一个函数应用于滚动窗口。我在这里看到的所有答案都集中在应用于单行/列，但我想将我的功能应用于整个窗口。这是一个简化的示例：

import pandas as pd
data = [ [1,2], [3,4], [3,4], [6,6], [9,1], [11,2] ]
df = pd.DataFrame(columns=list('AB'), data=data)

这是df：

    A   B
0   1   2
1   3   4
2   3   4
3   6   6
4   9   1
5   11  2

取一些函数应用于整个窗口：

df.rolling(3).apply(lambda x: x.shape)

在这个例子中，我想得到类似的东西：

    some_name   
0   NA  
1   NA  
2   (3,2)   
3   (3,2)   
4   (3,2)   
5   (3,2)   

当然，这里以形状为例说明f将整个窗口作为计算对象，而不仅仅是一行/列。我尝试使用axis关键字 forrolling以及raw关键字 forapply但没有成功。其他方法 ( agg, transform) 似乎也没有提供。

当然，我可以通过列表理解来做到这一点。只是认为有一种更简单/更清洁的方法来做到这一点。

Answer 1

不是pd.DataFrame.rolling.... 该函数被迭代地应用于列，接收一系列浮点数/NaN，并一个接一个地返回一系列浮点数/NaN。我认为你的直觉会更好......

def rolling_pipe(dataframe, window, fctn):
    return pd.Series([dataframe.iloc[i-window: i].pipe(fctn) 
                      if i >= window else None 
                      for i in range(1, len(dataframe)+1)],
                     index = dataframe.index) 

df.pipe(rolling_pipe, 3, lambda x: x.shape)