当前位置:首页 > 行业动态 > 正文

pandas rolling 计算最后一条数据

在Pandas中,我们可以使用rolling函数来计算滚动窗口内的统计量,如果我们想要获取滚动窗口内的最后一条数据,我们可以使用apply函数来应用一个自定义的函数。

以下是一个例子:

我们创建一个Pandas DataFrame:

import pandas as pd
创建一个DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3, 4, 5],
    'B': [6, 7, 8, 9, 10]
})

我们定义一个函数来获取滚动窗口内的最后一条数据:

def get_last(window):
    return window[1]

接下来,我们使用rolling函数创建一个滚动窗口,并使用apply函数来应用我们的自定义函数:

创建一个滚动窗口,窗口大小为2
rolling = df.rolling(window=2)
使用apply函数来应用我们的自定义函数
result = rolling.apply(get_last)

我们打印结果:

print(result)

这将输出以下结果:

     A     B
0  NaN   NaN
1  2.0   7.0
2  3.0   8.0
3  4.0   9.0
4  5.0  10.0

在这个例子中,我们可以看到,对于前两个元素,由于滚动窗口的大小为2,所以没有最后一条数据,因此结果是NaN,从第三个元素开始,滚动窗口内有最后一条数据,所以我们可以得到正确的结果。

0

随机文章