pandas rolling 计算最后一条数据
- 行业动态
- 2024-04-14
- 2730
在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,从第三个元素开始,滚动窗口内有最后一条数据,所以我们可以得到正确的结果。
本站发布或转载的文章及图片均来自网络,其原创性以及文中表达的观点和判断不代表本站,有问题联系侵删!
本文链接:http://www.xixizhuji.com/fuzhu/286088.html