量化交易开发之循环、多股策略语法(六)
一、用list数据类型存储多个股票
以如下这个简单的策略为例,学习在策略中操作多个股票:
def initialize(context):
run_daily(period, time='every_bar')
g.security = '000001.XSHE'
def period(context):
order(g.security, 100)
事实上,根据前面所学,我们是可以写多个股票的策略的,无非是把原来单个股票的操作类似的再写几遍:
def initialize(context):
run_daily(period, time='every_bar')
g.security1 = '000001.XSHE'
g.security2 = '000001.XSHE'
def period(context):
order(g.security1, 100)
order(g.security2, 100)
显然的问题是,当股票比较多的时候,就要写很多遍。因此我们要学习其他的写法,使用list数据类型存储多个股票,如下:
def initialize(context):
run_daily(period, time='every_bar')
# 把两个股票代码作为list存入g.security中
g.security = ['000001.XSHE', '000002.XSHE']
二、循环语句
循环语句for可以遍历任何序列的项目,比如一个list,一般用法如下:
# 含义是依次把序列中的元素赋给for后的变量,并执行循环语句
for 变量 in 一个序列:
要循环的语句,也叫循环体
下面我们来看个for的例子:
for k in ['David','richard', 'kars', 'yarid']:
print(k)
执行后,结果如下:
分析过程,for语句运行过程是,取出list中第一个元素’David’,并将其赋值给k,然后执行print()打印k。依次类推,直到最后一个元素打印完毕。
三、range()语句
使用for语句时有一个常见一起使用的语句range(),它的功能是生成等差数列的,用法如下:
range(首项,上限,步长)
# 首项 就是这个数列的第一项,可省略,省略后默认为0
# 步长 就是数列的公差、间隔,可省略,省略后默认为1
# 上限 是用来限制数列长度的,即数列不得大于或等于上限。不可省略。
# 另外,python2中range产生的是list,但python3中产生的不是list,但可以用list()这个语句把结果转成list类型,比如list(range(1,7,2))。我们策略编辑环境是python2。
# 一个例子
for j in range(1,7,2):
print(j)
print(range(1,7,2))
四、continue和break
continue与break是重要的修饰循环执行流程的语句,用法如下:
# break的作用是写在循环体中用来跳出当前的整个循环过程
# continue的作用是写在循环体中用来跳出当前的这一次的循环过程
# 通过一个例子应该就能明白两者的作用与区别
# 一个简单的循环例子
for t in range(4):
print(t)
# 执行的结果是
# 0
# 1
# 2
# 3
# 在例子中使用break。可以看到当循环到2的时候,打印omg后,执行break,终 了整个循环过程,不再继续循环3了,所以omg后就什么都没了。
for t in range(4):
if t == 2:
print('omg')
break
print(t)
# 在例子中使用continue。可以看到当循环到2的时候,打印omg后,执行continue,跳过了当前正循环的t为2这个循环过程的余下部分,不在继续执行之后的语句(即print(t),此时t等于2),而继续循环3了,所以omg后有打印3。
for t in range(4):
if t == 2:
print('omg')
continue
print(t)
continue:
break:
五、学以致用,写一个多股票策略
用刚学的知识把之前简单的策略例子改写成多股票版本,如下:
def initialize(context):
run_daily(period,time='every_bar')
# 把两个股票代码作为list存入g.security中
g.security = ['000001.XSHE','000002.XSHE']
def period(context):
for stk in g.security:
order(stk,100)
# 获得股票持仓成本
cost=context.portfolio.positions[stk].avg_cost
# 获得股票现价
price=context.portfolio.positions[stk].price
# 计算收益率
ret=price/cost-1
# 如果收益率小于-0.01,即亏损达到1%则卖出股票,幅度可以自己调一般10%
if ret<-0.01:
order_target(stk,0)
print('触发止损')
六、未完待续
下章将继续介绍综合我们所学写的自己写一个真实且完整的策略。
欢迎关注知乎:北京不北
欢迎+V:beijing_bubei
欢迎关注douyin:near.X (北京不北)
获得免费答疑,长期技术交流。
``