In [1]:
%matplotlib inline
%time from hikyuu.interactive.interactive import *
iodog.open()
#use_draw_engine('echarts')
std::cout are redirected to python::stdout
std::cerr are redirected to python::stderr
[2019-06-16 17:43:03.751] [trace] SQLITE3: c:\stock/stock.db
[2019-06-16 17:43:03.753] [info] Loading market information...
[2019-06-16 17:43:03.756] [info] Loading stock type information...
[2019-06-16 17:43:03.758] [info] Loading stock information...
[2019-06-16 17:43:06.617] [info] Loading KData...
[2019-06-16 17:43:06.624] [info] Preloading all day kdata to buffer!
[2019-06-16 17:43:15.068] [info] 8.45064s Loaded Data.
Wall time: 12.6 s

一、策略分析

原始描述

买入条件:周线MACD零轴下方底部金叉买入30%

卖出条件:日线级别 跌破 20日线 卖出50%持仓

策略分析

市场环境:无

系统有效性:无

信号指示器:

  • 买入信号:周线MACD零轴下方底部金叉,即周线的DIF>DEA金叉时买入(快线:DIF,慢线DEA)
  • 卖出信号:日线级别 跌破 20日均线

止损/止盈:无

资金管理:

  • 买入:30% (不明确,暂且当做当前现金的30%)
  • 卖出:已持仓股票数的50%

盈利目标:

移滑价差:

二、实现系统部件

定义信号指示器

In [2]:
def getNextWeekDate(week):
    """获取指定日期的下一周周一日期"""
    from datetime import timedelta
    py_week = week.datetime()
    next_week_start = py_week + timedelta(days = 7 - py_week.weekday())
    return Datetime(next_week_start)
In [3]:
def DEMO_SG(self):
    """
    买入信号:周线MACD零轴下方底部金叉,即周线的DIF>DEA金叉时买入
    卖出信号:日线级别 跌破 20日均线
    
    参数:
    week_macd_n1:周线dif窗口
    week_macd_n2: 周线dea窗口
    week_macd_n3: 周线macd平滑窗口
    day_n: 日均线窗口
    """
    k = self.getTO()
    if (len(k) == 0):
        return
    
    stk = k.getStock()
    
    #-----------------------------
    #计算日线级别的卖出信号        
    #-----------------------------
    day_c = CLOSE(k)
    day_ma = MA(day_c, self.getParam("day_n"))
    day_x = day_c < day_ma  #收盘价小于均线
    for i in range(day_x.discard, len(day_x)):
        if day_x[i] >= 1.0:
            self._addSellSignal(k[i].datetime)

    #-----------------------------
    #计算周线级别的买入信号        
    #-----------------------------
    week_q = QueryByDate(k[0].datetime, k[-1].datetime.nextDay(), kType=Query.WEEK)
    week_k = k.getStock().getKData(week_q)
    
    n1 = self.getParam("week_macd_n1")
    n2 = self.getParam("week_macd_n2")
    n3 = self.getParam("week_macd_n3")
    m = MACD(CLOSE(week_k), n1, n2, n3)
    fast = m.getResult(0)
    slow = m.getResult(1)
    
    discard = m.discard if m.discard > 1 else 1
    for i in range(discard, len(m)):
        if (fast[i-1] < slow[i-1] and fast[i] > slow[i]):
            #当周计算的结果,只能作为下周一的信号
            self._addBuySignal(week_k[i].datetime.nextWeek())
         

定义资金管理策略

In [4]:
class DEMO_MM(MoneyManagerBase):
    """
    买入:30% (不明确,暂且当做当前现金的30%)
    卖出:已持仓股票数的50%
    """
    def __init__(self):
        super(DEMO_MM, self).__init__("MACD_MM")
        
    def _reset(self):
        pass
    
    def _clone(self):
        return DEMO_MM()
        
    def _getBuyNumber(self, datetime, stk, price, risk, part_from):
        tm = self.getTM()
        cash = tm.currentCash
        
        #可以不用考虑最小交易单位的问题,已经自动处理
        #num = int((cash * 0.3 // price // stk.atom) * stk.atom)
        return int(cash*0.3/price) #返回类型必须是int
    
    def _getSellNumber(self, datetime, stk, price, risk, part_from):
        tm = self.getTM()
        position = tm.getPosition(stk)
        total_num = position.number
        num = int(total_num * 0.5)
        return num if num >= 100 else 0

三、构建并运行系统

设定系统参数

In [5]:
#账户参数
init_cash = 100000 #账户初始资金
init_date = Datetime('1990-1-1') #账户建立日期

#信号指示器参数
week_n1 = 12
week_n2 = 26
week_n3 = 9
day_n = 20

#选定标的,及测试区间
stk = sm['sz000001']
start_date = Datetime('2017-01-01')  #如果是同一级别K线,可以使用索引号,使用了不同级别的K线数据,建议还是使用日期作为参数
end_date = Datetime()

构建系统实例

In [6]:
#创建账户
my_tm = crtTM(datetime=init_date, initCash = init_cash, costFunc=TC_FixedA())

#创建系统实例
my_sys = SYS_Simple()

#绑定账户
my_sys.tm = my_tm

#绑定信号指示器
my_sys.sg = crtSG(DEMO_SG, 
              {'week_macd_n1': week_n1, 'week_macd_n2': week_n2, 'week_macd_n3': week_n3, 'day_n': day_n}, 
                'DEMO_SG') 
my_sys.sg.setParam('alternate', False)

#绑定资金管理策略
my_sys.mm = DEMO_MM()

运行系统

In [7]:
iodog.close()
q = QueryByDate(start_date, end_date, kType=Query.DAY)
my_sys.run(stk, q)

#将交易记录及持仓情况,保存在临时目录,可用Excel查看
#临时目录一般设置在数据所在目录下的 tmp 子目录
#如果打开了excel记录,再次运行系统前,记得先关闭excel文件,否则新的结果没法保存
my_tm.tocsv(sm.tmpdir())

四、查看资金曲线及绩效统计

In [8]:
#绘制资金收益曲线(净收益)
x = my_tm.getProfitCurve(stk.getDatetimeList(q), KQuery.DAY)
#x = my_tm.getFundsCurve(stk.getDatetimeList(q), KQuery.DAY) #总资产曲线
x = PRICELIST(x)
x.plot()
In [9]:
#回测统计
per = Performance()
print(per.report(my_tm, Datetime.now()))
帐户初始金额: 100000.00
累计投入本金: 100000.00
累计投入资产: 0.00
累计借入现金: 0.00
累计借入资产: 0.00
累计红利: 0.00
现金余额: 98258.95
未平仓头寸净值: 2198.24
当前总资产: 100457.19
已平仓交易总成本: 0.00
已平仓净利润总额: 0.00
单笔交易最大占用现金比例%: 29.68
交易平均占用现金比例%: 29.45
已平仓帐户收益率%: 0.00
帐户年复合收益率%: 0.22
帐户平均年收益率%: 0.22
赢利交易赢利总额: 0.00
亏损交易亏损总额: 0.00
已平仓交易总数: 0.00
赢利交易数: 0.00
亏损交易数: 0.00
赢利交易比例%: 0.00
赢利期望值: 0.00
赢利交易平均赢利: 0.00
亏损交易平均亏损: 0.00
平均赢利/平均亏损比例: 0.00
净赢利/亏损比例: 0.00
最大单笔赢利: 0.00
最大单笔亏损: 0.00
赢利交易平均持仓时间: 0.00
赢利交易最大持仓时间: 0.00
亏损交易平均持仓时间: 0.00
亏损交易最大持仓时间: 0.00
空仓总时间: 762.00
空仓时间/总时间%: 100.00
平均空仓时间: 762.00
最长空仓时间: 761.00
最大连续赢利笔数: 0.00
最大连续亏损笔数: 0.00
最大连续赢利金额: 0.00
最大连续亏损金额: 0.00
R乘数期望值: 0.00
交易机会频率/年: 0.00
年度期望R乘数: 0.00
赢利交易平均R乘数: 0.00
亏损交易平均R乘数: 0.00
最大单笔赢利R乘数: 0.00
最大单笔亏损R乘数: 0.00
最大连续赢利R乘数: 0.00
最大连续亏损R乘数: 0.00

五、或许想看下图形

In [10]:
my_sys.plot()
MA(CLOSE(my_sys.getTO()), 20).plot(new=False)

六、或许想看看所有股票的情况

In [11]:
import pandas as pd
def calTotal(blk, q):
    per = Performance()
    s_name = []
    s_code = []
    x = []
    for stk in blk:
        my_sys.run(stk, q)
        per.statistics(my_tm, Datetime.now())
        s_name.append(stk.name)
        s_code.append(stk.market_code)
        x.append(per.get("当前总资产".encode('gb2312')))
    return pd.DataFrame({'代码': s_code, '股票': s_name, '当前总资产': x})

%time data = calTotal(blocka, q)
Wall time: 14.5 s
In [12]:
#保存到CSV文件
#data.to_csv(sm.tmpdir() + '/统计.csv')
data[:10]
Out[12]:
代码 股票 当前总资产
0 SZ000532 华金资本 101028.21
1 SH601777 力帆股份 100000.00
2 SZ300238 冠昊生物 100000.00
3 SH600781 辅仁药业 100082.54
4 SH600770 综艺股份 100000.00
5 SH600671 天目药业 100298.20
6 SZ002888 惠威科技 102349.67
7 SH600086 东方金钰 98657.58
8 SH603599 广信股份 99178.59
9 SH600363 联创光电 99928.59
In [14]:
iodog.close()
my_tm.buy(Datetime(201906160000), sm['sz000001'], 10, 10000)
Out[14]:
<Trade(+infinity, , , UNKNOWN, 0.0000, 0.0000, 0.0000, 0.0000, 0, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000, --)>
In [ ]: