0

0

Pandas中实现带重置条件与最小窗口的滚动均值信号生成

心靈之曲

心靈之曲

发布时间:2026-02-05 17:43:00

|

599人浏览过

|

来源于php中文网

原创

Pandas中实现带重置条件与最小窗口的滚动均值信号生成

本文介绍如何基于列数据动态生成±1信号:当当前值达到滚动均值的指定倍数时翻转信号,并确保每次重置后滚动窗口至少累积指定最小长度才允许下一次触发。

在时间序列分析或量化信号处理中,常需构造“自适应状态指示器”——其值(如 1 或 -1)随数据局部统计特征动态切换。本教程解决一个典型场景:对 DataFrame 的每列独立计算带重置逻辑的滚动均值信号,满足以下核心约束:

  • 初始信号为 1;
  • 每次重置由当前值 ≥ multiple × 当前滚动窗口均值 触发;
  • 重置仅在滚动窗口已累积至少 min_count 个观测值后才被允许;
  • 一旦重置发生,累计和(acc)与计数(num_obs)清零,从下一行为起点重建滚动窗口;
  • 信号翻转(1 ↔ -1),并持续保持直至下一次有效重置。

该逻辑无法直接通过 pandas.DataFrame.rolling().mean() 实现,因其依赖状态感知的前向迭代(即当前决策影响后续窗口起始),而非无状态滑动计算。因此,我们采用显式循环 + 累积状态管理的方式,兼顾可读性与性能。

✅ 基础实现(纯 Python + Pandas)

import pandas as pd

def rolling_mean_signal(col, start=1, multiple=2, min_count=4):
    """
    生成滚动均值触发的±1信号序列

    Parameters:
    -----------
    col : pd.Series
        输入列数据
    start : int, default 1
        初始信号值(1 或 -1)
    multiple : float, default 2
        触发重置的倍数阈值
    min_count : int, default 4
        允许重置所需的最小窗口长度

    Yields:
    -------
    int : 当前行对应的信号值(1 或 -1)
    """
    curr = start
    num_obs = 0
    acc = 0.0

    for v in col:
        acc += v
        num_obs += 1

        if num_obs < min_count:
            yield curr
            continue

        mean_val = acc / num_obs
        if v >= multiple * mean_val:
            curr *= -1
            num_obs = 0
            acc = 0.0

        yield curr

# 示例应用
df = pd.DataFrame({
    "A": [0.1, 0.1, 0.15, 0.1, 0.1, 0.7, 0.1, 0.1, 0.5, 1, 0.1, 0.1],
    "B": [0.1, 0.1, 0.4, 0.1, 0.8, 0.1, 0.1, 0.1, 0.1, 0.1, 0.9, 0.1],
})

df["signal_A"] = list(rolling_mean_signal(df["A"]))
df["signal_B"] = list(rolling_mean_signal(df["B"]))
print(df[["A", "B", "signal_A", "signal_B"]])
? 关键点说明: acc 和 num_obs 是跨行维持的状态变量,模拟“动态滚动窗口”的累积过程; 重置后 num_obs=0 表示窗口清空,下一行将作为新窗口的第 1 个观测; yield 实现惰性生成,内存友好,适用于大数据流。

⚡ 高性能优化(Numba 加速)

对万行级以上数据,纯 Python 循环可能成为瓶颈。使用 numba.njit 可实现 JIT 编译加速(通常提升 5–20 倍):

秘塔写作猫
秘塔写作猫

秘塔写作猫是一个集AI写作、校对、润色、配图等为一体的创作平台

下载
from numba import njit
import numpy as np

@njit
def rolling_mean_signal_numba(col, start=1, multiple=2, min_count=4):
    n = len(col)
    out = np.empty(n, dtype=np.int8)

    curr = np.int8(start)
    num_obs = 0
    acc = 0.0

    for i in range(n):
        v = col[i]
        acc += v
        num_obs += 1

        if num_obs < min_count:
            out[i] = curr
            continue

        mean_val = acc / num_obs
        if v >= multiple * mean_val:
            curr *= -1
            num_obs = 0
            acc = 0.0

        out[i] = curr

    return out

# 应用加速版(注意传入 .values)
df["signal_A_fast"] = rolling_mean_signal_numba(df["A"].values)
df["signal_B_fast"] = rolling_mean_signal_numba(df["B"].values)

⚠️ 注意事项

  • Numba 不支持 Pandas 对象,务必传入 .values(np.ndarray);
  • dtype=np.int8 足够表示 ±1,节省内存;
  • 首次调用会触发编译,后续调用即达峰值性能。

? 总结与扩展建议

  • 该方案本质是在线状态机:每步输入一个值,输出当前状态,并更新内部状态;
  • 可轻松扩展为多级阈值(如 v >= 3*m 触发 curr *= -2)、滞后重置(延迟 1 行生效)或混合条件(如同时检查标准差);
  • 若需支持 NaN 安全处理,可在循环内添加 if np.isnan(v): ... 分支;
  • 对多列批量处理,推荐用 df.apply(lambda s: pd.Series(rolling_mean_signal(s))) 封装,保持接口一致性。

通过结合清晰的状态逻辑与可选的底层加速,本方法在可维护性与执行效率间取得良好平衡,适用于实时信号生成、异常检测触发及规则引擎开发等场景。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

72

2025.12.04

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

3

2026.01.31

if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

794

2023.08.22

lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

208

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

191

2025.11.08

Python lambda详解
Python lambda详解

本专题整合了Python lambda函数相关教程,阅读下面的文章了解更多详细内容。

58

2026.01.05

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

1258

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

275

2025.10.17

java连接字符串方法汇总
java连接字符串方法汇总

本专题整合了java连接字符串教程合集,阅读专题下面的文章了解更多详细操作。

4

2026.02.05

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 3.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.4万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号