将滑动窗口的聚合函数只应用于其中的一个列,可以使用pandas库中的rolling方法,然后选择需要聚合的列进行操作。
代码示例:
import pandas as pd
# 创建数据
data = {"col1": [1, 2, 3, 4, 5], "col2": [10, 20, 30, 40, 50]}
# 转化为dataframe
df = pd.DataFrame(data)
# 选择col2列,计算滑动窗口内的平均值
rolling_mean = df["col2"].rolling(window=2).mean()
print(rolling_mean)
输出:
0 NaN
1 15.0
2 25.0
3 35.0
4 45.0
Name: col2, dtype: float64
在上述示例中,我们使用了DataFrame的rolling方法,指定了当前得到的窗口大小为2,然后对第二列数据进行滑动计算并返回每个窗口中的平均值。