pandas rolling 计算最后一条数据

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

pandas rolling 计算最后一条数据
(图片来源网络,侵删)

以下是一个例子:

我们创建一个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,从第三个元素开始,滚动窗口内有最后一条数据,所以我们可以得到正确的结果。

原创文章,作者:未希,如若转载,请注明出处:https://www.kdun.com/ask/473792.html

(0)
未希未希新媒体运营
上一篇 2024-04-14 16:27
下一篇 2024-04-14 16:28

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

云产品限时秒杀。精选云产品高防服务器,20M大带宽限量抢购  >>点击进入