python - 如何在没有聚合函数的情况下使用 pandas 滚动函数？

Question

我有一个类似的数据框：

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

我想在 pandas 中使用滚动函数，但滚动后不需要聚合函数（sum、mean、min、max、...）。我需要每个窗口的数据。

我可以自己开发一个滑动窗口，但我的问题是关于滚动功能。通过在熊猫中使用滚动功能是否可以有类似的东西。

当我使用以下功能时

w = df.rolling(2)

我收到以下结果：

Rolling [window=2,center=False,axis=0]

虽然我需要这个：

w = [[0 , 1]
    [1 , 2]
    [2 , 3]
    [3 , 4]]

标签： pythonpandas

Answer 1

由于熊猫1.1滚动对象是可迭代的，所以你可以这样做：

w = list(df.rolling(2))