【Python】：利用rolling和apply函数的向下取值

最新推荐文章于 2024-07-24 17:22:27 发布

原创最新推荐文章于 2024-07-24 17:22:27 发布 · 986 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

Python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何在Python中结合使用rolling和apply函数对DataFrame数据进行滚动窗口操作，实现向下取值等统计分析任务。

import pandas as pd

def get_under_rolling(df,window,user,name):
    df[name] = df[user].iloc[::-1].rolling(window=window).apply(lambda x:x[0]).iloc[::-1]
    return df

if __name__ == '__main__':
    df = pd.DataFrame({'a':[1,2,3,4,5],
                   'b':[2,3,4,5,6]})
    # 把b列向下取值作为新的c列
    df = get_under_rolling(df, window=3, user='b',name='c')