Py学习  »  Python

别再让 AI 帮你选股了!手把手教你用 Python 搭建量化策略工厂

数据科学实战 • 1 周前 • 77 次点击  

 

 

 

 

 

 

 

 

 

 

 

 

 

2026年重磅升级已全面落地!欢迎加入专注财经数据与量化投研的【数据科学实战】知识星球!您将获取持续更新的《财经数据宝典》与《量化投研宝典》,双典协同提供系统化指引;星球内含300篇以上独有高质量文章,深度覆盖策略开发、因子分析、风险管理等核心领域,内容基本每日更新;同步推出的「量化因子专题教程」系列(含完整可运行代码与实战案例),系统详解因子构建、回测与优化全流程,并实现日更迭代。我们持续扩充独家内容资源,全方位赋能您的投研效率与专业成长。无论您是量化新手还是资深研究者,这里都是助您少走弯路、事半功倍的理想伙伴,携手共探数据驱动的投资未来!

引言

你是不是也这样用 AI 做交易?打开 ChatGPT,问一句"明天大盘怎么走?"——然后根据回答决定买卖。

醒醒吧,这不是量化,这是算命。

真正的量化机构从不依赖某一个"神奇指标"或"完美策略"。他们的核心武器是一台策略生产机器——能够持续不断地生成、测试、筛选和替换策略。这台机器才是真正的竞争优势。

今天这篇文章,我们就基于一篇来自 Ztrader 的深度研究文章,手把手教你用 Python 搭建一个属于自己的 AI 量化策略工厂(Strategy Factory)。不需要博士学位,不需要昂贵的数据终端,只需要 Python 和一颗愿意动手的心。


一、为什么你需要一个"策略工厂"?

大多数散户的交易流程是这样的:

找到一个策略 → 情感绑定 → 硬撑 → 爆仓

而机构的流程是这样的:

策略发现 → 部署 → 衰退 → 退役 → 替换

核心区别在于:策略是有生命周期的。市场结构会变化,流动性会迁移,波动率机制会轮换,你精心调优的参数很可能只是运气好。

所以目标不是"找到一个永远有效的策略",而是:

建立一个能比策略失效速度更快地生产新策略的系统。

这个系统就是你的策略工厂,其核心流程如下:

交易品种选定 → 策略构思 → 代码生成 → 回测 → 指标计算 → 筛选过滤 → 策略库


二、第 1 步:定义交易品种(别什么都想做)

新手最常犯的错误就是"什么都想测"——股票、期权、期货、外汇、加密货币全都要。这样你什么都建不起来。

从一个精简的品种池开始。对于初学者,推荐:

  • • SPY:大盘权益 Beta
  • • QQQ:科技/风险偏好 Beta
  • • GLD:宏观对冲
  • • TLT:利率代理、机制信号

品种越少 = 迭代越快 + 自由度越低。这才是通向"可部署"研究的路径。


三、第 2 步:用 AI 生成策略候选,而非预测

大多数人问 AI 的问题是错的:"市场接下来往哪走?"这是娱乐,不是研究。

正确的做法是让 AI 输出可编程、可测试的交易规则。你可以使用这个提示词模板:

请为 {资产} 生成 10 个策略候选方案。
限定策略家族为:动量 / 均值回归 / 突破。
每个候选方案必须包含:
1. 精确的入场规则(可编程)
2. 精确的出场规则(可编程)
3. 参数范围(不是单一值)
4. 预期失效场景(何时会失灵)
以 JSON 格式输出。

记住:你的工作不是"相信" AI,而是让 AI 为你生产研究输入,然后你来测试验证。


四、第 3 步:标准化策略结构(告别意大利面条代码)

要做工厂化生产,每个策略都必须接入统一接口。每个策略应该被简化为三个函数:

  1. 1. signal():生成交易信号(做多/做空/空仓)
  2. 2. position():仓位管理
  3. 3. risk():风险控制

即使你一开始不实现第 2 和第 3 个函数,也要把架构设计好。因为没有标准化,你就无法批量测试、无法公平比较、无法自动化筛选,你的"研究"最终只会变成一堆乱七八糟的脚本。


五、第 4 步:搭建最小回测引擎

你不需要复杂的框架就能开始。你需要的是一个你能信任且能扩展的工具。

首先安装依赖:

pip install yfinance pandas numpy

下面是最小化交易引擎的完整代码:

import numpy as np
import
 pandas as pd
import
 yfinance as yf

# 一年交易日天数

TRADING_DAYS = 252

def
 load_data(symbol, start="2015-01-01"):
    """加载行情数据并计算日收益率"""

    df = yf.download(symbol, start=start)
    df = df[["Open", "High", "Low", "Close", "Volume"]].dropna()
    df["ret"] = df["Close"].pct_change()  # 计算每日收益率
    return
 df.dropna()

def
 backtest(df, signal, fee_bps=1.0):
    """
    回测函数
    signal: 交易信号序列(1=做多,-1=做空,0=空仓)
    fee_bps: 手续费(基点)
    """

    # 信号向后移一位,避免前瞻偏差

    pos = signal.shift(1).fillna(0)
    # 计算毛收益

    gross = pos * df["ret"]
    # 手续费代理:仓位变化时扣费

    turnover = pos.diff().abs().fillna(0)
    fee = turnover * (fee_bps / 10000.0)
    # 计算净收益

    net = gross - fee
    # 计算权益曲线

    equity = (1 + net).cumprod()
    return
 net, equity

def
 metrics(strategy_ret):
    """计算策略核心指标:总收益、最大回撤、夏普比率"""

    equity = (1 + strategy_ret.fillna(0)).cumprod()
    dd = equity / equity.cummax() - 1  # 回撤序列
    sharpe = (
        np.sqrt(TRADING_DAYS)
        * strategy_ret.mean()
        / (strategy_ret.std() + 1e-12)
    )
    total_return = float(equity.iloc[-1] - 1)
    max_dd = float(dd.min())
    return
 {
        "total_return"
: total_return,   # 总收益
        "max_dd"
: max_dd,               # 最大回撤
        "sharpe"
: float(sharpe),        # 夏普比率
        "trades_proxy"
: int(strategy_ret.ne(0).sum())  # 交易次数估算
    }

这段代码故意写得很"无聊"。因为:无聊 = 可审计,可审计 = 可靠,可靠 = 可部署。


六、第 5 步:构建策略模板族

大多数盈利策略都是几个经典家族的变体。先从两个最基础的开始:

A)趋势/动量策略——移动平均线

def signal_ma_trend(df, n=50):
    """
    移动平均线趋势策略
    价格在均线之上做多,之下做空
    n: 移动平均线周期
    """

    ma = df["Close"].rolling(n).mean()
    sig = pd.Series(0, index=df.index)
    sig[df["Close"] > ma] = 1   # 价格高于均线,做多
    sig[df["Close"] < ma] = -1  # 价格低于均线,做空
    return
 sig

B)均值回归策略——RSI 指标

def signal_rsi_mr(df, n=14, buy=30, sell=70):
    """
    RSI 均值回归策略
    RSI 低于 buy 阈值做多,高于 sell 阈值做空
    n: RSI 计算周期
    buy: 超卖阈值
    sell: 超买阈值
    """

    delta = df["Close"].diff()
    gain = delta.clip(lower=0)               # 涨幅
    loss = -delta.clip(upper=0)              # 跌幅
    rs = gain.rolling(n).mean() / (loss.rolling(n).mean() + 1e-12)
    rsi = 100 - (100 / (1 + rs))            # 计算 RSI 值
    sig = pd.Series(0, index=df.index)
    sig[rsi < buy] = 1    # RSI 超卖,做多
    sig[rsi > sell] = -1   # RSI 超买,做空
    return
 sig

有了两个策略家族之后,工厂的"DNA"就位了。


七、第 6 步:生成参数网格

一个策略不是"MA50",而是一整个参数空间。真正的研究就发生在参数网格的遍历中:

def grid_ma():
    """生成移动平均线策略的参数网格"""

    return
 [{"n": n} for n in [10, 20, 30, 50, 100, 200]]

def
 grid_rsi():
    """生成 RSI 策略的参数网格"""

    grid = []
    for
 n in [7, 14, 21]:         # RSI 周期
        for
 buy in [20, 30]:      # 超卖阈值
            for
 sell in [70, 80]: # 超买阈值
                grid.append({"n": n, "buy": buy, "sell": sell})
    return
 grid

八、第 7 步:批量回测 + 排名(策略工厂核心循环)

现在把所有环节串起来,构建完整的工厂循环:

def


    
 run_factory(symbol="SPY"):
    """
    策略工厂主函数
    批量回测所有策略家族的所有参数组合,筛选并排名
    """

    df = load_data(symbol)
    results = []

    # 遍历移动平均线趋势策略家族

    for
 p in grid_ma():
        sig = signal_ma_trend(df, n=p["n"])
        ret, _ = backtest(df, sig, fee_bps=1.0)
        m = metrics(ret)
        results.append({"family": "MA_TREND", "params": p, **m})

    # 遍历 RSI 均值回归策略家族

    for
 p in grid_rsi():
        sig = signal_rsi_mr(df, n=p["n"], buy=p["buy"], sell=p["sell"])
        ret, _ = backtest(df, sig, fee_bps=1.0)
        m = metrics(ret)
        results.append({"family": "RSI_MR", "params": p, **m})

    out = pd.DataFrame(results)

    # 基础筛选:最大回撤不超过 30%,夏普比率大于 0.7

    out = out[(out["max_dd"] > -0.30) & (out["sharpe"] > 0.7)]

    # 按夏普比率和总收益降序排名

    out = out.sort_values(["sharpe", "total_return"], ascending=False)
    return
 out

if
 __name__ == "__main__":
    ranked = run_factory("SPY")
    print
(ranked.head(10))  # 打印排名前 10 的策略

运行这段代码,你就完成了:策略生成、参数探索、回测、筛选、排名。这就是一个策略工厂的基线版本


九、第 8 步:防过拟合——简单的滚动前推测试

AI 非常擅长帮你"优化"出一条漂亮的权益曲线——但那可能只是精心拟合历史数据的幻觉。你需要一个"杀手开关"。

最简单的版本:

  • • 训练集:2015–2020 年
  • • 测试集:2021 年至今

如果策略只在训练集上表现良好,那它就是垃圾。你不需要博士学位,你需要的是纪律


实战案例:SPY 策略工厂输出

在实际运行中,你的输出会是这样的画面:

  • • 大部分参数组合表现平庸
  • • 少数"还不错"
  • • 极少数在不同子时段内表现稳定

重点不是追求完美,而是建立一个可重复的流程——发现当前有效的策略,替换掉失效的策略。这就是机构的生存之道。


总结

读完这篇文章,你应该明白了一个核心理念:策略是消耗品,系统才是资产。

大多数散户拥有的是指标、观点和"感觉"。而你现在可以拥有的是:

  • • 一套完整的研究工作流
  • • 一个可扩展的回测引擎
  • • 一套筛选和排名纪律
  • • 一个可持续迭代的策略库

从今天开始,不要再问 AI "明天买什么"了。用 AI 生成策略候选,用 Python 构建回测引擎,用数据做你的裁判。

市场不再是情绪的战场,而是你的数据集。


参考文章

加入专注于财经数据与量化投研的知识星球【数据科学实战】,获取本文完整研究解析、代码实现细节。

财经数据与量化投研知识社区

2026年全面升级已落地!【数据科学实战】知识星球核心权益如下:

  1. 1. 双典系统赋能:获赠《财经数据宝典》与《量化投研宝典》完整文档,凝练多年实战经验,构建系统化知识框架;
  2. 2. 量化因子日更教程(2026重磅新增):每日更新「量化因子专题教程」,配套完整可运行代码与实战案例,深度拆解因子构建、回测与优化全流程;
  3. 3. 量化文章专题教程库:300+篇星球独有高质量教程式文章,系统覆盖策略开发、因子研究、风险管理等核心领域,内容基本每日更新,并配套精选学习资料与实战参考;
  4. 4. PyBroker实战课程:赠送《PyBroker-入门及实战》视频课程,手把手教学,快速掌握量化策略开发技能;
  5. 5. 财经数据支持:定期更新国内外财经数据,为策略研发提供精准、可靠的数据基础;
  6. 6. 顶尖学者与行业专家分享:年度邀请学术界博士与业界资深专家开展前沿论文精讲与实战案例分享,不少于4场,直击研究前沿与产业实践;
    专家直连答疑:与核心开发者及领域专家实时互动,高效解决投研实战难题;
  7. 7. 专业社群与专属福利:加入高质量交流社群,获取课程折扣及更多独家资源。

星球已沉淀丰富内容生态——涵盖量化文章专题教程库、因子日更系列、高频数据集、PyBroker实战课程、专家深度分享与实时答疑服务。无论您是初探量化的学习者,还是深耕领域的从业者,这里都是助您少走弯路、高效成长的理想平台。诚邀加入,共探数据驱动的投资未来!

 

好文推荐

1. 用 Python 打造股票预测系统:Transformer 模型教程(一)

2. 用 Python 打造股票预测系统:Transformer 模型教程(二)

3. 用 Python 打造股票预测系统:Transformer 模型教程(三)

4. 用 Python 打造股票预测系统:Transformer 模型教程(完结)

5. 揭秘隐马尔可夫模型:因子投资的制胜武器

6. YOLO 也能预测股市涨跌?计算机视觉在股票市场预测中的应用

7. 金融 AI 助手:FinGPT 让你轻松掌握市场分析

8. 量化交易秘籍:为什么专业交易员都在用对数收益率?

9. Python 量化投资利器:Ridge、Lasso 和 Elastic Net 回归详解

10. 掌握金融波动率模型:完整 Python 实现指南



Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/194231