量化交易开发之context与判断（五）
2024-01-29 本文已影响0人
北京不北
量化交易开发之context与判断（五）
通过前文的讲解，我们已经能理解最开始那个最简单的策略例子，如下：
def initialize(context):
run_daily(period, time = 'every_bar')
g.security = '000001.XSHE'
def period(context):
order(g.security, 100)
下面我们看看什么是context。
一、context的结构
-
context是一个回测系统建立的Context类型的对象，其中存储了如当前策略运行的时间点、所持有的股票、数量、持仓成本等数据。
-
对象可以理解为特殊类型的变量，对象的结构往往比我们之前见过的list与dict更复杂，被定义好的对象是有名字的，比如context是一个变量，它的变量类型是一个Context类型的对象，就像dict包括key与value，Context类型的对象也包括很多属性，而且可以嵌套另一个种类型的对象。
1706520776061.png
二、数据取用方法
- 获取对象类型变量内包含的数据方法是用英文句号隔开，而当包含的是另一个对象时，只需在应用英文句号隔开即可，例子如下：
# 打印可用的资金
print(context.portfolio.available_cash)
# 打印运行的频率
print(context.run_params.frequency)
# 打印当前时间的开始时间
print(context.current_dt)
# 执行后日志内容如下
# 1000000.0
# day
# 2016-06-01 09:30:00
1706520776061.png
- 当要获取的对象内的数据是另一种有结构的变量类型时，比如dict或者list，正常按照该变量类型进一步取用数据即可。例如context.portfolio.posiotions是一个dct，例子如下，这次给出完整代码：
# context.portfolio.positions的含义是仓位信息，需要在取之前买入并持有股票
def initialize(context):
run_daily(period, time='every_bar')
g.security = '000001.XSHE'
def period(context):
order(g.security, 100)
# 打印所有键
print(context.portfolio.positions.keys())
# 打印所有值
print(context.portfolio.positions.values())
# 打印g.security的开仓均价
print(context.portfolio.positions[g.security].avg_cost)
# 执行后日志内容如下
# ['000001.XSHE']
# [UserPosition({'avg_cost': 8.539999999999997, 'security': '000001.XSHE', 'closeable_amount': 0, 'price': 8.53, 'total_amount': 100})]
# 8.54
常用的context数据写法如下，推荐自己动手试一下：
- 当前时间 context.current_dt
- 当前时间的“年-月-日”的字符串格式 context.current_dt.strftime("%Y-%m-%d")
- 前一个交易日 context.previous_date
- 当前可用资金 context.portfolio.available_cash
- 持仓价值 context.portfolio.positions_value
- 累计收益 context.portfolio.returns
- 当前持有股票 context.portfolio.positions.keys()
- 当前持有的某股票的开仓均价 context.portfolio.positions['xxxxxx.xxxx'].avg_cost
- 当前持有的某股票的可卖持仓量 context.portfolio.positions['xxxxxx.xxxx'].closeable_amount
三、条件判断
能够获取context的数据后，我们会考虑利用这些数据丰富策略的逻辑，但在此之前我们还要学习if条件判断语句，如下：
# 如果 条件1成立为 True 将执行代码块1
# 如果 条件1不成立为False，将判断条件2
# 如果 条件2成立为 True 将执行代码块2
# 如果 条件2还不成立为False，将执行代码块3
if 条件1:
代码块1
elif 条件2:
代码块2
else:
代码块3
# 注意
# elif 可以有多个连续写
# 且elif和else都可以省略
# 条件判断语句中可以嵌套条件判断语句
举几个例子：
# 打印a、b中最大值
if a>=b:
print(a)
else:
print(b)
# 判断a的正负性
if a>0:
print('正')
elif a<0:
print('负')
elif a==0:
print('零')
# 如果当前是2018-05-04，则下单买入100股平安银行
date=context.current_dt.strftime("%Y-%m-%d")
if date=='2018-05-04':
order('000001.XSHE',100)
# 判断a大小情况
if a>0:
if a<1:
print('a大于0且小于1')
else:
print('a大于等于1')
else:
print('a小于等于0')
条件判断语句比较简单，但还需说明的是条件的写法中用到的运算符：
# 写条件常用运算符：
# < 小于
# > 大于
# <= 小于等于
# >= 大于等于
# == 等于
# != 不等于
# and 与，即and两边条件同为真，则真
# or 或，即or两边条件任意一个为真，则真
# not 非，即not右侧条件为真，则假，not右侧条件为假，则真
# 以判断a是否为0的几个写法为例
# 写法1
if a!=0:
print('否')
else:
print('是')
# 写法2
if a>0 or a<0:
print('否')
else:
print('是')
# 写法2
if a>=0 and a=<0:
print('是')
else:
print('否')
# 写法3
if not a==0:
print('否')
else:
print('是')
四、止损
- 狭义的止损是指当亏损达到一定幅度后下单卖出该股票的操作，目的是减少进一步的亏损。广义则指在狭义的思路上衍生的复杂的减少亏损的方法。
- 更多的情况下指狭义的止损。综合运用前文的讲过的内容我们已经可以实现当亏损达到一定幅度后下单卖出该股票的止损操作了。
通过context的数据可以得到持有股票的成本和现价，从而可以算出该股票的盈亏情况，运用条件判断语句根据盈亏情况从而决定是否卖出股票，从而实现止损操作，代码如下：
def initialize(context):
run_daily(period,time='every_bar')
g.security = '000001.XSHE'
def period(context):
# 买入股票
order(g.security, 100)
# 获得股票持仓成本
cost=context.portfolio.positions['000001.XSHE'].avg_cost
# 获得股票现价
price=context.portfolio.positions['000001.XSHE'].price
# 计算收益率
ret=price/cost-1
# 打印日志
print('成本价：%s' % cost)
print('现价：%s' % price)
print('收益率：%s' % ret)
# 如果收益率小于-0.01，即亏损达到1%则卖出股票，幅度可以自己调，一般10%
if ret<-0.01:
order_target('000001.XSHE',0)
print('触发止损')
- 设置回测时间为从2017-03-01到2017-03-31，初始资金为100000，频率为天。回测发现会在2017-03-20触发止损。
五、未完待续
下章将继续介绍context中的数据和条件判断。
欢迎关注知乎：北京不北
欢迎+V：beijing_bubei
欢迎关注douyin：near.X （北京不北）
获得技术答疑，长期技术交流。
