开源量化交易平台zipline初探

通过之前的Quantopian系列文章,大家已经了解Quantopian是一个在线构建量化交易策略的平台,而zipline则是Quantopian开源的python量化交易库,提供了Quantopian大部分的功能(如回测、研究)。
本文就带大家快速的搭建一个zipline环境。

可以通过pip或者conda安装zipline,这里推荐大家使用conda, 因为conda同时提供了环境与包的管理,使用起来更为方便。

安装Anaconda

Anaconda 是包含了conda、python及常用可用计算库的一个一体化的安装包,同时也提供了一个UI界面供使用。

下载地址:
https://www.anaconda.com/download/#linux

windows直接下载exe安装,linux用户下载后直接执行该shell

$ chmod +x Anaconda3-5.2.0-Linux-x86_64.sh
$ ./Anaconda3-5.2.0-Linux-x86_64.sh 

同时由于国内网速的原因,推荐切换至清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

确认是否安装成功

$ conda -V
conda 4.5.4

安装zipline

首先使用 conda创建一个单独的环境供使用,可指定python版本,在环境中安装的包不会影响其他环境

# 构建名为env_zipline的环境,指定python版本为3.5
$ conda create -n env_zipline python=3.5

切换至该环境

$ source activate env_zipline

可以看到shell前缀已经显示在该环境中

(env_zipline) $ 

使用conda安装zipline

# -c 表示指定channel
(env_zipline) $ conda install -c Quantopian zipline

数据准备

在写算法策略前,我们需要一些数据,本文通过Quandl API 来获取,你可以去网站免费注册一个账号(获取API_KEY),并使用如下命令将数据灌入zipline。

$ QUANDL_API_KEY=<yourkey> zipline ingest [-b <bundle>]

其中<bundle> 是数据包的名称,默认为 quandl

国内速度感人,可以自己想想办法 o( ̄︶ ̄)o,提示如下

$ QUANDL_API_KEY=api_key proxychains4 zipline ingest -b quandl

可参考之前的番前篇

在执行zipline ingest时,处理进程莫名被killed,最后通过系统日志/var/log/*发现为OOM(Out of memory)被系统杀死,可调大虚拟机内存/swap,或调整OOM-Killer配置。

进程被杀

经常会失败,可能需要清理 data bundles

$ zipline clean [-b <bundle>] --keep-last 0 #数字表示保留最近的n次数据

编写算法策略

每个算法策略包含两个函数

  • initialize(context)
  • handle_data(context, data)

在算法执行前,zipline首先会调用 initialize() 传入contextcontext是算法执行过程中共用的上下文信息,可以用来存储算法执行不同阶段的需要保存的数据。
初始化后,zipline会根据事件调用 handle_data(),每次调用,都会传入contextdatacontext是执行过程中唯一的一份,但data则与该次调用的事件相关,通常会包含此刻的开盘,最高、最低及收盘价。

来尝试写一个最简单的算法策略

from zipline.api import order, record, symbol

def initialize(context):
    pass

def handle_data(context, data):
    order(symbol('AAPL'), 10)
    record(AAPL=data.current(symbol('AAPL'), 'price'))

首先需要引入使用的函数,所使用的函数大部分来自zipline.api。当前算法中我们使用了orderrecord两个函数,order有两个参数——证券对象和买卖数量(为正标识买,为负表示卖),本例中我们买了10股苹果的股票。
record函数用来记录每次迭代过程中的你希望保存的数据,使用key=value的方式设置,当算法执行完后你就可以通过key来获取保存的数据。

执行算法策略

zipline提供了三种执行算法策略的方式:

  1. 命令行
  2. Jupyter Notebook
  3. run_algorithm()函数

命令行

首先我们使用命令行来执行下:

zipline run -f ../../zipline/examples/buyapple.py --start 2016-1-1 --end 2018-1-1 -o buyapple_out.pickle

run首先会调用initialize函数,然后使用handle_data()对历史股票价格数据逐天处理,每调用一次,本策略就会买入10股苹果的股票。每次调用order()时,zipline就会在委托订单中新增10股苹果股票,handle_data执行结束后,zipline就会去寻找尚未成交的委托,并尝试去成交,如果交易数量巨大,通常需要考虑佣金及滑点的影响(简单来说,就是你会花费比股价*数量更多的钱)。

现在我们来看看结果,结果类型为pandas中的Dataframe。

import pandas as pd
perf = pd.read_pickle('buyapple_out.pickle') # read in perf DataFrame
perf.head()
结果Dataframe

可以使用matplotlib将苹果股价与我们策略的运行结果绘制出来。

%pylab inline
figsize(12, 12)
import matplotlib.pyplot as plt

ax1 = plt.subplot(211)
perf.portfolio_value.plot(ax=ax1)
ax1.set_ylabel('Portfolio Value')
ax2 = plt.subplot(212, sharex=ax1)
perf.AAPL.plot(ax=ax2)
ax2.set_ylabel('AAPL Stock Price')
图示

因为我们的策略中只是单纯的买入苹果股票,可以看到两图基本保持一致。

Jupyter Notebook

Jupyter Notebook为python解释器提供了非常强大的浏览器界面,现在业界很多的quant都在使用,在notebook中也可以方便的执行zipline。
在notebook中,需要先加载zipline,然后使用魔术方法%%zipline来执行zipline命令,参数与命令行完全相同。

%load_ext zipline

%%zipline --start 2016-1-1 --end 2018-1-1
from zipline.api import symbol, order, record

def initialize(context):
    pass

def handle_data(context, data):
    order(symbol('AAPL'), 10)
    record(AAPL=data[symbol('AAPL')].price)

在notebook中我们就不在需用单独定义一个文件,会自动使用当前cell中的算法,同时,输出也不在需要单独定义,可以使用_.head() 直接获取。

至此,我们就完整搭建了一套zipline的环境,后续就可以随意玩耍了。

感谢阅读,欢迎订阅~


量化投资
配图
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,117评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,963评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,897评论 0 240
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,805评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,208评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,535评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,797评论 2 311
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,493评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,215评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,477评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,988评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,325评论 2 252
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,971评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,807评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,544评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,455评论 2 266

推荐阅读更多精彩内容