如何使用Python api 函数写股票策略

写策略需要了解的语法包括两方面，一方面是语言本身的语法（包括相关库），另一方面是量化平台提供的api。量化平台提供的api帮助文件里都有了,本文主要介绍写策略经常用到的库（datetime、numpy、pandas）中的一些函数。

1、 利用datetime库做日期、时间操作

利用context.now可以获得当前策略运行的时间，返回的是datetime.datetime格式。datetime.datetime格式可以很方便的进行日期、时间操作。比如timedelta可以很方便的在日期上做日、小时、分钟、秒的运算。例如，需要策略运行时间1天前的时间，可以这样写：context.now+datetime.timedelta(days=-1)，返回的便是一天前的时间。

2、 利用pandas做数据变频。

量化平台一般只提供分钟或者日频的数据，如果我们需要周、月的数据怎么办呢？pandas的resample函数可以很好的解决这个问题。举个例子，假如我们需要沪深300每月最后一个交易日的收盘价，我们可以这样写：

d= get_history(100,'1d','close')[‘000300.SH’].resample(‘m’，how=’last’)

3、利用set格式选取交集

有的时候我们利用不同标准会得到不同的股票池，如果想得到不同股票池的共同股票，那么可以使用set格式。Set格式可以很方便的进行交集、并集等集合运算。这里举一个例子，比如我们通过设置一定的财务数据条件会得到一个dataframe，记作df。df的列为股票代码，但可能并不都是我们所需的，比如我们只想得到沪深300的成份股，那么可以这样操作：1、s=get_index_constituents(‘000300.SH’) 得到沪深300的成份股。2、z=set(s) & set(df.columns) 得到交集z。3、得到的z是set格式，需要转换为list格式，可以这样操作 zl=list(z)。4、利用pandas 函数 筛选出我们需要的个股 d=df.loc[:zl] d即位我们所需要的数据。

4、 利用try…except跳过出错部分代码

有时候我们会遇到一些不是很重要的问题，但是由于遇到这类问题会报错，从而影响程序执行，这时我们希望的是忽略这些错误就可以了。下面举一个例子说明try…except 的用法。

比如：我们用p表示一只股票某一时刻的价格，v表示这只股票的成交量，我们想计算p/v，但是有可能该股票没有成交量（停牌，或者涨跌停了）这时直接计算就会出错，程序会跳出。这是我们可以try…except做如下处理：

try:d=p/v

except:d=0

意思就是如果计算p/v发生错误，那么就将d赋值为0

5、 利用pickle模块保存变量

有时候我们需要将当前工作空间的变量保存下来，这时可以使用pickle来解决，举个例子，假如目前工作空间有a,b,c三个变量，我们可以进行如下操作：1、f=open(‘file.pkl’,’wb’) 建立一个文件链接。2、import pickle;pickle.dump([a,b,c],f) 这样就可以保存了。读取时1、f=open(‘file.pkl’,’rb’) 2、import pickle;z=pickle.load([a,b,c],f) 这样将读取的变量都保存到了z变量中，通过z[0],z[1],z[2]可以获取a,b,c变量。