
使用 python 的量化平台目前很多啊,现在就三家**ricequant,joinquant,raquant**做一下简单评测
同样一段双均线( SMA 长短线)策略,虽然这个双均线,没必要每分钟都刷,毕竟作为超短期指标,双均线意义不大。
所以这也是让策略开发者困扰的一点,有时候有些策略的逻辑在“被选择的频度”下面毫无意义。镭矿则没有这一困扰,镭矿策略代码里面规定是什么频度执行就是什么频度执行。
言归正传,现在我们来用这个经典策略对比一下各个平台,分钟频度下一年半(2015-05-01 到 2017-01-01 )的回测速度。每分钟刷日线数据肯定是更没有意义,所以小编将 ricequant 和 joinquant 的获取历史数据的参数手动改为获取分钟数据
策略 A: 回测时间: 2015-05-01 至 2017-01-01 使用分钟数据,短均线上穿长均线,则买入,反之,则清仓卖出 对篇幅很恐惧的同学请先看一下这个表格:
策略 A 的回测时间对比表格
下面是镭矿 raquant的代码,耗时 17 秒
def init(context): context.s1="sha-600000" sma12_factor=SMAFactor(12,"close") sma30_factor=SMAFactor(30,"close") reg_factor("sma12",sma12_factor) reg_factor("sma30",sma30_factor) def every_minute(context,data): stock=context.s1 ma12=factor_output("sma12",context.s1,"m1")["sma12"] ma30=factor_output("sma30",context.s1,"m1")["sma30"] if ma12>ma30: order(stock,1000) elif ma12 < ma30 and context.portfolio.positions[stock].amount > 0: order_target_value(stock,0) 下面是 joinquant 的代码,耗时 6 分钟
# 初始化函数,设定要操作的股票、基准等等 def initialize(context): # 定义一个全局变量, 保存要操作的股票 # 000001(股票:平安银行) g.security = '000001.XSHE' # 设定沪深 300 作为基准 set_benchmark('000300.XSHG') # 每个单位时间(如果按天回测,则每天调用一次,如果按分钟,则每分钟调用一次)调用一次 def handle_data(context, data): security = g.security close_data = attribute_history(security, 10, '1m', ['close'],df=False) ma5 = close_data['close'][-5:].mean() ma10 = close_data['close'].mean() cash = context.portfolio.cash if ma5 > ma10: order_value(security, cash) log.info("Buying %s" % (security)) elif ma5 < ma10 and context.portfolio.positions[security].closeable_amount> 0: order_target(security, 0) log.info("Selling %s" % (security)) # 绘制五日均线价格 record(ma5=ma5) # 绘制十日均线价格 record(ma10=ma10) 下面是 ricequant 的代码,耗时 3 到 4 分钟,貌似比 joinquant 的快,也有可能是因为这个例程的买入条件比较多?但是明显 ricequant 的回测准备时间较长。
import talib # 在这个方法中编写任何的初始化逻辑。 context 对象将会在你的算法策略的任何方法之间做传递。 def init(context): context.s1 = "000001.XSHE" # 设置这个策略当中会用到的参数,在策略中可以随时调用,这个策略使用长短均线,我们在这里设定长线和短线的区间,在调试寻找最佳区间的时候只需要在这里进行数值改动 context.SHORTPERIOD = 20 context.LOnGPERIOD= 120 # 你选择的证券的数据更新将会触发此段逻辑,例如日或分钟历史数据切片或者是实时数据切片更新 def handle_bar(context, bar_dict): prices = history(context.LONGPERIOD+1, '1m', 'close')[context.s1].values short_avg = talib.SMA(prices, context.SHORTPERIOD) long_avg = talib.SMA(prices, context.LONGPERIOD) plot("short avg", short_avg[-1]) plot("long avg", long_avg[-1]) cur_position = context.portfolio.positions[context.s1].quantity # 计算现在 portfolio 中的现金可以购买多少股票 shares = context.portfolio.cash/bar_dict[context.s1].close # 如果短均线从上往下跌破长均线,也就是在目前的 bar 短线平均值低于长线平均值,而上一个 bar 的短线平均值高于长线平均值 if short_avg[-1] - long_avg[-1] < 0 and short_avg[-2] - long_avg[-2] > 0 and cur_position > 0: # 进行清仓 order_target_value(context.s1, 0) # 如果短均线从下往上突破长均线,为入场信号 if short_avg[-1] - long_avg[-1] > 0 and short_avg[-2] - long_avg[-2] < 0: # 满仓入股 order_shares(context.s1, shares) 日线级别上,策略 A 回测时间对比表格(镭矿需要手动修改函数名 every_minute 为 every_day) 回测时间 镭矿 秒回(少于一秒) ricequant 5 秒后秒回。(感觉任何回测都需要准备 5 秒钟) joinquant 秒回(少于一秒)
对比不明显,我们来升级一下策略逻辑。为了简便,我们仍然修改策略 A ,形成策略 B 。
事实上这不是一个真正的策略,不产生任何交易。只不过我们这里为了尽快知道各个平台的回测速度罢了。
策略 B 每日轮询 50 只股票的 SMA 长短线, record 出符合 SMA 短线上穿长线的股票个数 镭矿 raquant的代码
def init(context): context.stocks=find_by_group('sz50') sma12_factor=SMAFactor(12,"close") sma30_factor=SMAFactor(30,"close") reg_factor("sma12",sma12_factor) reg_factor("sma30",sma30_factor) def every_day(context,data): cnt=0 for stock in context.stocks: ma12=factor_output("sma12",stock)["sma12"] ma30=factor_output("sma30",stock)["sma30"] if ma12>ma30: cnt=cnt+1 record("cnt",cnt) ricequant 代码
import talib # 在这个方法中编写任何的初始化逻辑。 context 对象将会在你的算法策略的任何方法之间做传递。 def init(context): context.stocks =concept('央企 50') context.SHORTPERIOD = 20 context.LOnGPERIOD= 120 def handle_bar(context, bar_dict): cnt=0 for stock in context.stocks: prices = history(context.LONGPERIOD+1, '1d', 'close')[stock].values short_avg = talib.SMA(prices, context.SHORTPERIOD) long_avg = talib.SMA(prices, context.LONGPERIOD) if short_avg[-1] - long_avg[-1] < 0 and short_avg[-2] - long_avg[-2] > 0: cnt=cnt+1 plot("cnt",cnt) joinquant 的代码:
# 初始化函数,设定要操作的股票、基准等等 def initialize(context): # 定义一个全局变量, 保存要操作的股票 # 000001(股票:平安银行) g.stocks =get_concept_stocks('GN177') set_benchmark('000300.XSHG') # 每个单位时间(如果按天回测,则每天调用一次,如果按分钟,则每分钟调用一次)调用一次 def handle_data(context, data): cnt=0 # 获取股票的收盘价 for stock in g.stocks: close_data = attribute_history(stock, 10, '1d', ['close'],df=False) # 取得过去五天的平均价格 ma5 = close_data['close'][-5:].mean() # 取得过去 10 天的平均价格 ma10 = close_data['close'].mean() # 取得当前的现金 if ma5>ma10: cnt=cnt+1 record(cnt=cnt) 你们一定很好奇这次得对比结果,所以小编故意卖关子放到了最后
日线级别上,策略 B 回测时间对比表格